Python tf.device用法及代碼示例

指定在此上下文中創建/執行的操作的設備。

用法

tf.device(
    device_name
)

此函數指定用於在特定上下文中創建/執行的操作的設備。嵌套上下文將繼承並在指定設備上創建/執行其操作。如果不需要特定設備，請考慮不使用此函數，以便自動分配設備。通常，此函數的使用是可選的。 device_name 可以完全指定，如“/job:worker/task:1/device:cpu:0”，或部分指定，僅包含"/" 分隔字段的子集。指定的任何字段都將覆蓋來自外部範圍的設備注釋。

例如：

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'