我的大小张量[无,4],我想将其切成二维,以给我一个尺寸[无]张量。在numpy中,这将是t [:,dimension],是否有一种快速的方法可以使用TensorFlow命令?
您可以使用与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])