tf.device

2022-09-03 22:03:57 浏览数 (2)

为在此上下文中创建/执行的操作系统指定设备。

代码语言:javascript复制
tf.device(device_name)

device_name可以完全指定,如“/job:worker/task:1/device:cpu:0”,或者部分指定,只包含“/”分隔字段的子集。指定的任何字段都覆盖来自外部范围的设备注释。例如:

代码语言:javascript复制
with tf.device('/job:foo'):
  # ops created here have devices with /job:foo
  with tf.device('/job:bar/task:0/device:gpu:2'):
    # ops created here have the fully specified device above
  with tf.device('/device:gpu:1'):
    # ops created here have the device '/job:foo/device:gpu:1'

参数:

  • device_name:要在上下文中使用的设备名称。

返回值:

  • 上下文管理器,指定新创建的操作系统使用的默认设备。

可能产生的异常:

  • RuntimeError: If a function is passed in.

0 人点赞