Python tensorflow.DeviceSpec.device_index属性用法及代码示例

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

device_index用于获取DeviceSpec的设备索引。

用法：tensorflow.DeviceSpec.device_index

返回值：它返回设备索引。

范例1：

Python3

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

输出：


Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba96aee8>
Device Index: 0

范例2：

Python3

# Importing the library 
import tensorflow as tf 
  
# Initializing Device Specification 
device_spec = tf.DeviceSpec(job ="gfg", device_type ="CPU", device_index = 1) 
  
# Printing the DeviceSpec object 
print('Device Spec:', device_spec) 
  
# Getting device index 
device_index = device_spec.device_index 
  
# Printing the result 
print('Device Index:', device_index)

输出：


Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba97f048>
Device Index: 1

相关用法

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