Python – tensorflow.DeviceSpec

TensorFlow是谷歌设计的开源Python库，用于开发机器学习模型和深度学习神经网络。

DeviceSpec代表TensorFlow设备的规范。这个规范可能是部分的。如果一个DeviceSpec被部分指定，它将根据其定义的范围与其他DeviceSpec`s合并。

语法: tensorflow.DeviceSpec( job, replica, task, device_type, device_index )

参数:

• job（可选）:它是一个字符串，用于指定工作名称。
• replica（可选）:它是一个整数，指定副本的索引。
• task（可选）:它是一个整数，指定了任务的索引。
• device_type（可选）:它可以是CPU或GPU。
• device_index（可选）:它是一个整数，指定了设备的索引。

返回：它返回一个DeviceSpec对象。

示例 1:

# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", device_type ="GPU", device_index = 0)
  
# Printing the result
print('DeviceSpec: ', device_spec)

输出:

DeviceSpec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5c1818ac8>

示例 2:

# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", device_type ="CPU", device_index = 0)
  
# Printing the result
print('DeviceSpec: ', device_spec)

输出:

DeviceSpec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5bb29d888>