如何在张量流中切片数组

Question

我有一个大小为[None，4]的张量T，我想沿第二维对其进行切片以给我一个大小为[None]的张量。在numpy中，这将是T [:, DIMENSION]，有没有一种快速的方法来使用tensorflow命令？

Answer 1

您可以使用与numpy相同的操作。

a = tf.constant([[1,2,3,4],[5,6,7,8],[7,8,9,0]])
a.shape

a的形状是 Out[]:TensorShape([Dimension(3), Dimension(4)])

使用切片操作，获取第二维的值：

a[:, 2].eval()

，输出为

array([3, 7, 9])