當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python tensorflow.DeviceSpec.task屬性用法及代碼示例

TensorFlow是Google設計的開源Python庫,用於開發機器學習模型和深度學習神經網絡。

task用於獲取DeviceSpec的任務索引。

用法:tensorflow.DeviceSpec.task

返回值:它返回DeviceSpec的任務索引

範例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)

輸出:

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

範例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)

輸出:

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





相關用法


注:本文由純淨天空篩選整理自aman neekhara大神的英文原創作品 Python – tensorflow.DeviceSpec.task Attribute。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。