def _test(self, x, p): xtf = tf.constant(x) val_true = stats.geom.logpmf(x, p) with self.test_session(): self.assertAllClose(geom.logpmf(xtf, p).eval(), val_true) self.assertAllClose( geom.logpmf(xtf, tf.constant(p)).eval(), val_true)
def _test(x, p): xtf = tf.constant(x) val_true = stats.geom.logpmf(x, p) _assert_eq(geom.logpmf(xtf, p), val_true) _assert_eq(geom.logpmf(xtf, tf.constant(p)), val_true) _assert_eq(geom.logpmf(xtf, tf.constant([p])), val_true)
def _test(self, x, p): val_true = stats.geom.logpmf(x, p) with self.test_session(): self.assertAllClose(geom.logpmf(x, p=p).eval(), val_true)
def _test(self, x, p): xtf = tf.constant(x) val_true = stats.geom.logpmf(x, p) with self.test_session(): self.assertAllClose(geom.logpmf(xtf, p).eval(), val_true) self.assertAllClose(geom.logpmf(xtf, tf.constant(p)).eval(), val_true)