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]])
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)
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)
def _reduce_batch_count(x): return tf_utils.reduce_batch_count( x, reduce_instance_dims=reduce_instance_dims)