Python tensorflow.IndexedSlices.device屬性用法及代碼示例

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

device用於查找將在其上生成值的設備的名稱。

用法：tensorflow.IndexedSlices.device

返回值：它返回設備的名稱。

範例1：

Python3

# Importing the library 
import tensorflow as tf 
  
# Initializing the input 
data = tf.constant([1, 2, 3]) 
  
# Printing the input 
print('data:', data) 
  
# Calculating result 
res = tf.IndexedSlices(data, [0]) 
  
# Finding device name  
device = res.device 
  
# Printing the result 
print('device:', device)

輸出：


data: tf.Tensor([1 2 3], shape=(3, ), dtype=int32)
device: /job:localhost/replica:0/task:0/device:CPU:0

範例2：

Python3

# Importing the library 
import tensorflow as tf 
  
# Initializing the input 
data = tf.constant([[1, 2, 3], [4, 5, 6]]) 
  
# Printing the input 
print('data:', data) 
  
# Calculating result 
res = tf.IndexedSlices(data, [0]) 
  
# Finding device name  
device = res.device 
  
# Printing the result 
print('device:', device)

輸出：


data: tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
device: /job:localhost/replica:0/task:0/device:CPU:0

相關用法

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