TensorFlow保持形状相同时切片?

Question

我想把一个元素从一个维度中取出来，同时保持形状不变。

张量的形状为:(BATCH_SIZE, N_STEPS, NUM_FEATURES)我想创建一个新的张量(BATCH_SIZE, 1, NUM_FEATURES)，其中1是最后一步。

输入张量形状为(None, 128,16)我尝试创建一个新的张量，如下所示:

X = X[:,-1,:]

X的形状变成了(None, 16)，但我需要这个是(None, 1,16)

Answer 1

更新:我得到这个工作与以下代码:

s = tf.shape(X)
X = tf.reshape(X[:,-1,:],shape=[s[0],1,s[2]])