Python tf.DeviceSpec.parse_from_string用法及代码示例

用法

parse_from_string(
    spec
)

将 DeviceSpec 名称解析为其组件。

2.x 行为改变：

在 TensorFlow 1.x 中，此函数会改变自己的状态并返回自己。在 2.x 中，DeviceSpec 是不可变的，该函数将返回一个包含该规范的 DeviceSpec。

• 推荐的：

  # my_spec and my_updated_spec are unrelated.
  my_spec = tf.DeviceSpec.from_string("/CPU:0")
  my_updated_spec = tf.DeviceSpec.from_string("/GPU:0")
  with tf.device(my_updated_spec):
    ...

• 将在 1.x 和 2.x 中工作(尽管在 2.x 中已弃用)：

  my_spec = tf.DeviceSpec.from_string("/CPU:0")
  my_updated_spec = my_spec.parse_from_string("/GPU:0")
  with tf.device(my_updated_spec):
    ...

• 不会在 2.x 中工作：

  my_spec = tf.DeviceSpec.from_string("/CPU:0")
  my_spec.parse_from_string("/GPU:0")  # <== Will not update my_spec
  with tf.device(my_spec):
    ...

一般来说，DeviceSpec.from_string应该完全替换DeviceSpec.parse_from_string，DeviceSpec.replace应该直接完全替换设置属性。