Skip to content
Related Articles

Related Articles

Improve Article

Python – tensorflow.DeviceSpec.task Attribute

  • Last Updated : 01 Aug, 2020
Geek Week

TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning  neural networks. 

task is used to get the task index of DeviceSpec.

Syntax: tensorflow.DeviceSpec.task

Returns: It returns the task index of DeviceSpec

Example 1:



Python3




# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg2", replica = 5, device_type ="GPU", device_index = 1)
  
# Printing the DeviceSpec object
print('Device Spec: ', device_spec)
  
# Getting task index
task = device_spec.task
  
# Printing the result
print('Task Index: ', task)

Output:

Device Spec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba981d08>
Task Index:  None

Example 2:

Python3




# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg2", replica = 5, task = 2, device_type ="GPU", device_index = 1)
  
# Printing the DeviceSpec object
print('Device Spec: ', device_spec)
  
# Getting task index
task = device_spec.task
  
# Printing the result
print('Task Index: ', task)

Output:

Device Spec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba91a708>
Task Index:  2


 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course




My Personal Notes arrow_drop_up
Recommended Articles
Page :