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


Python tensorflow.DeviceSpec.from_string()用法及代碼示例

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

from_string用於從字符串生成DeviceSpec。

用法:tensorflow.DeviceSpec.from_string( spec )

參數:

  • spec:它是/job:/副本:/task:/device:CPU:或/job:/replica:/task:/device:GPU形式的字符串,所有條目都是可選的。

返回值:它返回由字符串生成的DeviceSpec對象。



範例1:

Python3

# Importing the library 
import tensorflow as tf 
  
# Formatting the string 
spec = '/job:gfg / replica:1 / task:2'
  
# Initializing Device Specification 
device_spec = tf.DeviceSpec.from_string(spec) 
  
# Printing the DeviceSpec object 
print('Device Spec:', device_spec)

輸出:


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

範例2:

Python3

# Importing the library 
import tensorflow as tf 
  
# Formatting the string 
spec = '/job:gfg / replica:2 / task:3'
  
# Initializing Device Specification 
device_spec = tf.DeviceSpec.from_string(spec) 
  
# Printing the DeviceSpec object 
print('Device Spec:', device_spec)

輸出:


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

相關用法


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