tensor也可以作为索引

Mr.Jcak

于 2018-12-03 15:27:24 发布

阅读量4.3k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： TensorFlow 文章标签： TensorFlow tensor

本文链接：https://blog.youkuaiyun.com/weixin_38314865/article/details/84767220

TensorFlow 专栏收录该内容

26 篇文章

订阅专栏

本文探讨了在TensorFlow框架下，如何正确地使用Tensor作为索引。Tensor可以用于索引同样为Tensor类型的变量，但不能直接用于list类型的变量。通过示例展示了将list转换为Tensor后，如何进行有效的索引操作。

在TensorFlow中，tensor也可以作为索引，但只能作为同样为tensor类型变量的索引，不能作为list类型变量的索引如下面的例子：

import tensorflow as tf

index = tf.to_int32([0,1,2])  # index是一个tensor
a = [[1,2,3], [4,5,6]]
b = a[index[0]]
sess = tf.InteractiveSession()
print(b)

TypeError: list indices must be integers or slices, not Tensor

若将a变为tensor类型

import tensorflow as tf

index = tf.to_int32([0,1,2])  # index是一个tensor
a = tf.convert_to_tensor([[1,2,3], [4,5,6]])
b = a[index[0]]
sess = tf.InteractiveSession()
print(sess.run(b))  # [1 2 3]