TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
device_index is used to get the device index of DeviceSpec.
Syntax: tensorflow.DeviceSpec.device_index
Returns: It returns the device index.
Example 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)
|
Output:
Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba96aee8> Device Index: 0
Example 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)
|
Output:
Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba97f048> Device Index: 1