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()。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。