def testInvalidAxis(self):
with self.assertRaisesRegexp(ValueError,
r'Argument `axis` must be negative.*'):
tfb.Cumsum(axis=0, validate_args=True)
with self.assertRaisesRegexp(
TypeError, r'Argument `axis` is not an `int` type\.*'):
tfb.Cumsum(axis=-1., validate_args=True)
def testInvalidAxis(self):
with self.assertRaisesRegexp(ValueError,
"`axis` must be a negative integer."):
tfb.Cumsum(axis=0)
with self.assertRaisesRegexp(ValueError,
"`axis` must be a negative integer."):
tfb.Cumsum(axis=-1.)
def testInvalidAxis(self):
with self.assertRaisesOpError('Argument `axis` must be negative.'):
self.evaluate(
tfb.Cumsum(axis=0, validate_args=True).forward([0., 1.]))
with self.assertRaisesRegexp(
TypeError, r'Argument `axis` is not an `int` type\..*'):
tfb.Cumsum(axis=-1., validate_args=True)
def testBijectiveAndFinite(self):
bijector = tfb.Cumsum()
x = np.linspace(-10, 10, num=10).astype(np.float32)
y = np.cumsum(x, axis=-1)
bijector_test_util.assert_bijective_and_finite(bijector,
x,
y,
eval_func=self.evaluate,
event_ndims=1)
def _checkBijectorInAllDims(self, x):
"""Helper for `testBijector`."""
x = self._build_tensor(x)
for axis in range(-self.evaluate(tf.rank(x)), 0):
bijector = tfb.Cumsum(axis=axis)
self.assertStartsWith(bijector.name, "cumsum")
y = tf.cumsum(x, axis=axis)
self.assertAllClose(y, self.evaluate(bijector.forward(x)))
self.assertAllClose(x, self.evaluate(bijector.inverse(y)))
def _checkEqualTheoreticalFldj(self, x):
"""Helper for `testJacobian`."""
event_ndims = int(self.evaluate(tf.rank(x)) - 1)
self.assertGreaterEqual(event_ndims, 1)
bijector = tfb.Cumsum(axis=-event_ndims)
fldj = bijector.forward_log_det_jacobian(
self._build_tensor(x), event_ndims=event_ndims)
fldj_theoretical = bijector_test_util.get_fldj_theoretical(
bijector, x, event_ndims=event_ndims)
fldj_, fldj_theoretical_ = self.evaluate([fldj, fldj_theoretical])
self.assertAllEqual(np.zeros_like(fldj_), fldj_)
self.assertAllClose(np.zeros_like(fldj_theoretical_), fldj_theoretical_)
def testInvalidAxis(self):
with self.assertRaisesRegexp(ValueError,
"`axis` needs to be negative."):
tfb.Cumsum(axis=0)
with self.assertRaisesRegexp(TypeError, "`axis` is not an `int`."):
tfb.Cumsum(axis=-1.)
