Ejemplos de reduce_batch_count en Python

Ejemplo n.º 1

Archivo: tf_utils_test.py Proyecto: GeckoMuscle/transform

 def test_reduce_batch_count_elementwise(self):
     x = tf.constant([[[1], [2]], [[1], [2]]])
     with tf.compat.v1.Session():
         self.assertAllEqual(
             tf_utils.reduce_batch_count(x,
                                         reduce_instance_dims=False).eval(),
             [[2], [2]])

Ejemplo n.º 2

Archivo: tf_utils_test.py Proyecto: subodhchhabra/transform

 def test_reduce_batch_count(self):
     x = tf.constant([[[1], [2]], [[1], [2]]])
     with tf.Session():
         self.assertAllEqual(
             tf_utils.reduce_batch_count(x,
                                         reduce_instance_dims=True).eval(),
             4)

Ejemplo n.º 3

Archivo: tf_utils_test.py Proyecto: yuhonghong7035/transform

 def test_reduce_batch_count_sparse(self):
   x = tf.SparseTensor(
       indices=[[0, 0, 0], [0, 2, 0], [1, 1, 0], [1, 2, 0]],
       values=[1., 2., 3., 4.],
       dense_shape=[2, 4, 1])
   with tf.Session():
     self.assertAllEqual(
         tf_utils.reduce_batch_count(x, reduce_instance_dims=True).eval(), 4)

Ejemplo n.º 4

Archivo: tf_utils_test.py Proyecto: robertlugg/transform

 def _reduce_batch_count(x):
     return tf_utils.reduce_batch_count(
         x, reduce_instance_dims=reduce_instance_dims)