fwgrad_tests.py 文件源码

def test_basic(self):
    with tf.Graph().as_default(), self.test_session() as sess:
      rnd = np.random.RandomState(0)
      x = self.get_random_tensor([18, 12], rnd=rnd)
      y = tf.Print(x, [x])
      self.assert_bw_fw(sess, x, y, rnd=rnd)