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()。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。