如何在 Tensorflow 1.12 中从二维张量中获取张量的一部分?
How do I get a slice of Tensor from a 2D Tensor in Tensorflow 1.12?
我有一个维度为 (a,b) 的二维张量 X,我需要从中获取一维切片,是否可以在 TF 中执行类似于 numpy 的操作,例如:
slice = X[t,:]
如果您需要在 Keras/TensorFlow 模型中执行此操作,您可以按如下方式利用 lambda 层:
layer = tf.keras.layers.Lambda(lambda x: x[:,:,t])(input)
否则,您应该能够使用与 python 中相同的语法来执行此操作,如此处所述:TensorFlow API. Finally, you can also try the TensorFlow slice method tf.slice
我有一个维度为 (a,b) 的二维张量 X,我需要从中获取一维切片,是否可以在 TF 中执行类似于 numpy 的操作,例如:
slice = X[t,:]
如果您需要在 Keras/TensorFlow 模型中执行此操作,您可以按如下方式利用 lambda 层:
layer = tf.keras.layers.Lambda(lambda x: x[:,:,t])(input)
否则,您应该能够使用与 python 中相同的语法来执行此操作,如此处所述:TensorFlow API. Finally, you can also try the TensorFlow slice method tf.slice