Python – tensorflow.DeviceSpec.parse_from_string()
TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。
parse_from_string()用于将DeviceSpec名称解析为其组成部分。
语法: tensorflow.DeviceSpec.parse_from_string( spec )
参数:
- spec : 它是一个形式为/job:/replica:/task:/device:CPU:或/job:/replica:/task:/device:GPU的字符串,所有字段是可选的。
返回:它返回一个DeviceSpec对象。
示例 1:
# Importing the library
import tensorflow as tf
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", replica = 5)
# Printing the DeviceSpec
print('Device Spec: ', device_spec.to_string())
# Getting new DeviceSpec object
new_device_spec = device_spec.parse_from_string("/GPU:0")
# Printing the result
print('New Device Spec: ', new_device_spec.to_string())
输出:
Device Spec: /job:gfg/replica:5
New Device Spec: /device:GPU:0
示例 2:
# Importing the library
import tensorflow as tf
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", replica = 5)
# Printing the DeviceSpec
print('Device Spec: ', device_spec.to_string())
# Getting new DeviceSpec object
new_device_spec = device_spec.parse_from_string("replica:2 / GPU:0")
# Printing the result
print('New Device Spec: ', new_device_spec.to_string())
输出:
Device Spec: /job:gfg/replica:5
New Device Spec: /replica:2/device:GPU:0