TensorFlow 测试

由 Carrie 创建, 最后一次修改 2017-08-26

单元测试

TensorFlow 提供了一个方便的类,继承 unittest.TestCase,可以添加与 TensorFlow 测试相关的方法。如下述的例子:tf.

import tensorflow as tf

class SquareTest(tf.test.TestCase):

  def testSquare(self):
    with self.test_session():
      x = tf.square([2, 3])
      self.assertAllEqual(x.eval(), [4, 9])

if __name__ == '__main__':
  tf.test.main()

tf.test.TestCase 继承了 unittest.TestCase,但添加了一些额外的方法,我们将很快证明这些方法。

  • tf.test.main
  • tf.test.TestCase
  • tf.test.test_src_dir_path

TensorFlow 实用程序

  • tf.test.assert_equal_graph_def
  • tf.test.get_temp_dir
  • tf.test.is_built_with_cuda
  • tf.test.is_gpu_available
  • tf.test.gpu_device_name

TensorFlow 梯度检测

tf.test.compute_gradient 和 tf.test.compute_gradient_error 对图进行数值微分,以便与已注册的分析梯度进行比较。

以上内容是否对您有帮助:

二维码
建议反馈
二维码