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應該直接完全替換設置屬性。