def test_errors(self):
for o in [object(), None, 1.2, LONG_MAX,
tf.constant(1.2, dtype=tf.float32),
tf.constant(LONG_MAX, dtype=tf.int64)]:
with pytest.raises(
TypeError,
match='group_ndims cannot be converted to int32'):
_ = reduce_group_ndims(tf.reduce_sum, tf.constant(0.), o)
with pytest.raises(
ValueError, match='group_ndims must be non-negative'):
_ = reduce_group_ndims(tf.reduce_sum, tf.constant(0.), -1)
with self.test_session():
with pytest.raises(
Exception, match='group_ndims must be non-negative'):
_ = reduce_group_ndims(tf.reduce_sum, tf.constant(0.),
tf.constant(-1, dtype=tf.int32)).eval()
def test_output(self):
tensor = tf.reshape(tf.range(24, dtype=tf.float32), [2, 3, 4])
tensor_sum_1 = tf.reduce_sum(tensor, axis=-1)
tensor_sum_2 = tf.reduce_sum(tensor, axis=[-2, -1])
tensor_prod = tf.reduce_prod(tensor, axis=-1)
g0 = tf.constant(0, dtype=tf.int32)
g1 = tf.constant(1, dtype=tf.int32)
g2 = tf.constant(2, dtype=tf.int32)
with self.test_session():
# static group_ndims
np.testing.assert_equal(
tensor.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, 0).eval()
)
np.testing.assert_equal(
tensor_sum_1.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, 1).eval()
)
np.testing.assert_equal(
tensor_sum_2.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, 2).eval()
)
np.testing.assert_equal(
tensor_prod.eval(),
reduce_group_ndims(tf.reduce_prod, tensor, 1).eval()
)
# dynamic group_ndims
np.testing.assert_equal(
tensor.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, g0).eval()
)
np.testing.assert_equal(
tensor_sum_1.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, g1).eval()
)
np.testing.assert_equal(
tensor_sum_2.eval(),
reduce_group_ndims(tf.reduce_sum, tensor, g2).eval()
)
np.testing.assert_equal(
tensor_prod.eval(),
reduce_group_ndims(tf.reduce_prod, tensor, g1).eval()
)
def log_prob(self, given, group_ndims=0, name=None):
return reduce_group_ndims(tf.reduce_sum,
tf.convert_to_tensor(given) - 1.,
group_ndims)