如何从张量中提取子矩阵?

Question

我想提取张量的前 2 列。

例如 考虑这个张量

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

输出应为

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

我玩过tf.unstack((，但不能完全成功。

Answer 1

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

tensor[:,0:2]

如果你熟悉python的标准切片符号，你已经知道如何使用tensorflow的。

Answer 2

你可以使用 tf.slice((

tf.slice(varTensor, [0, 0], [2, 2]);

你可以测试它

import tensorflow as tf
tf.Session().run(tf.slice([[0,1,-1,-2,-3],[1,0,-9, -2, -3]], [0, 0], [2,2]))