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


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

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

IndexedSlices()用於查找給定索引處的一組張量切片的稀疏表示。

用法:tensorflow.IndexedSlices(values, indices, dense_shape = None)

參數:

  • values:它是任何dtype的張量。
  • indices:它是一維張量。

返回值:它返回一個IndexedSlices對象。



範例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]) 
  
# Printing the result 
print('res:', res)

輸出:


data: tf.Tensor([1 2 3], shape=(3, ), dtype=int32)
res: IndexedSlices(indices=[0], values=tf.Tensor([1 2 3], shape=(3, ), dtype=int32))

範例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]) 
  
# Printing the result 
print('res:', res)

輸出:


data: tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
res: IndexedSlices(indices=[0], values=tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32))




相關用法


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