Пример #1
0
    def test_scan(self):
        with self.test_session():
            set_seed(42)
            op = tf.scan(lambda a, x: a + x, tf.constant([2.0, 3.0, 1.0]))

            self.assertAllClose(op.eval(), [2.0, 5.0, 6.0])
            self.assertAllClose(copy(op).eval(), [2.0, 5.0, 6.0])
Пример #2
0
    def test_scan_random(self):
        with self.test_session() as session:
            set_seed(1234)
            op = tf.scan(lambda a, x: a + x, tf.random_normal([3]))
            copy_op = copy(op)

            result = session.run([copy_op, copy_op, op, op])
            self.assertAllClose(result[0], result[1])
            self.assertAllClose(result[2], result[3])
Пример #3
0
 def test_dict_rv_rv(self):
   with self.test_session():
     set_seed(325135)
     x = Normal(mu=0.0, sigma=0.1)
     y = tf.constant(1.0)
     z = x * y
     qx = Normal(mu=10.0, sigma=0.1)
     z_new = copy(z, {x: qx})
     self.assertGreater(z_new.eval(), 5.0)
Пример #4
0
 def test_dict_tensor_rv(self):
     with self.test_session():
         set_seed(95258)
         x = Normal(mu=0.0, sigma=0.1)
         y = tf.constant(1.0)
         z = x * y
         qx = Normal(mu=10.0, sigma=0.1)
         z_new = copy(z, {x.value(): qx})
         self.assertGreater(z_new.eval(), 5.0)
Пример #5
0
 def test_dict_rv_tensor(self):
   with self.test_session():
     set_seed(289362)
     x = Normal(mu=0.0, sigma=0.1)
     y = tf.constant(1.0)
     z = x * y
     qx = Normal(mu=10.0, sigma=0.1)
     z_new = copy(z, {x: qx.value()})
     self.assertGreater(z_new.eval(), 5.0)