如何在Tensorflow 1.12中从2D张量获取张量？

Question

我有一个尺寸为（a，b）的二维张量X，我需要从中获取一维切片，是否可以在TF中执行类似于numpy的操作，如：

slice = X[t,:]

Answer 1

如果需要在Keras / TensorFlow模型中执行此操作，则可以利用lambda层，如下所示：

layer = tf.keras.layers.Lambda(lambda x: x[:,:,t])(input)

否则，您应该能够使用与python中相同的语法来做到这一点，如此处报告的TensorFlow API。最后，您还可以尝试TensorFlow slice方法tf.slice