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