当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法

parse_from_string(
    spec
)

参数

  • spec 形式的可选字符串
    /job:<name>/replica:<id>/task:<id>/device:CPU:<id>
    或者
    /job:<name>/replica:<id>/task:<id>/device:GPU:<id>
    因为 cpu 和 gpu 是互斥的。所有条目都是可选的。

返回

  • DeviceSpec

抛出

  • ValueError 如果规范无效。

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_stringDeviceSpec.replace应该直接完全替换设置属性。

相关用法


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