コード例 #1
0
 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)
コード例 #2
0
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)
コード例 #3
0
ファイル: test_geom_logpmf.py プロジェクト: blei-lab/edward
 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)
コード例 #4
0
 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)            
コード例 #5
0
 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)