Python tf.test.TestCase.session用法及代码示例

用法

@contextlib.contextmanager
session(
    graph=None, config=None, use_gpu=True, force_gpu=False
)

参数

graph 在返回的会话期间使用的可选图表。
config 用于配置会话的可选 config_pb2.ConfigProto。
use_gpu 如果为 True，请尝试在 GPU 上运行尽可能多的操作。
force_gpu 如果为 True，则将所有操作固定到 /device:GPU:0 。

生成(Yield)

一个 Session 对象，应该用作上下文管理器来围绕测试用例中的图形构建和执行代码。

用于执行测试的 TensorFlow 会话的上下文管理器。

请注意，这会将这个会话和图形设置为全局默认值。

使用 use_gpu 和 force_gpu 选项来控制操作的运行位置。如果 force_gpu 为 True，则所有操作都固定到 /device:GPU:0 。否则，如果 use_gpu 为 True，TensorFlow 会尝试在 GPU 上运行尽可能多的操作。如果 force_gpu and use_gpu 都为 False，则所有操作都固定到 CPU。

例子：

class MyOperatorTest(test_util.TensorFlowTestCase):
  def testMyOperator(self):
    with self.session():
      valid_input = [1.0, 2.0, 3.0, 4.0, 5.0]
      result = MyOperator(valid_input).eval()
      self.assertEqual(result, [1.0, 2.0, 3.0, 5.0, 8.0]
      invalid_input = [-1.0, 2.0, 7.0]
      with self.assertRaisesOpError("negative input not supported"):
        MyOperator(invalid_input).eval()

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.test.TestCase.session。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。