python - 如何从张量中提取子矩阵？

Question

我想提取张量的前两列。

例如， 考虑这个张量

[[0, 1, -1, -2, -3],

 [1, 0, -8, -2, -3]]

输出应该是

[[0,1],
[1,0]]

我使用了 tf.unstack() 但无法完全成功。

Answer 1

使用tensorflow的python切片运算符，比 Kris Roofe 的解决方案更具可读性:

tensor[:,0:2]

如果您熟悉Python的标准切片表示法，那么您已经知道如何使用tensorflow。