📜  Python – tensorflow.IndexedSlices()

📅  最后修改于: 2022-05-13 01:55:37.442000             🧑  作者: Mango

Python – tensorflow.IndexedSlices()

TensorFlow 是由 Google 设计的开源Python库,用于开发机器学习模型和深度学习神经网络。

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)


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], 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))