python - 为什么 Tensorflow 转置在特定场景下失败？

Question

tf.reset_default_graph()

sess=tf.Session()

X=tf.placeholder(tf.float32, shape=[3,8])

X_1 = tf.reshape(X,[3,2,4])

print(X_1)

X_tensor = tf.transpose(X_1,[1,2,0])

print(X_tensor)

输出:

Tensor("Reshape:0", shape=(3, 2, 4), dtype=float32)

Tensor("transpose:0", shape=(2, 4, 3), dtype=float32)

转置结果应为(4,3,2)。

Answer 1

您将 perm=[1,2,0] 传递给 tf.transpose，这就是您得到 (2, 4, 3) 的原因> 作为 (3, 2, 4) 的形状。 (参见:https://www.tensorflow.org/api_docs/python/tf/transpose)

尝试 X_tensor = tf.transpose(X_1,[2,0,1]) 以获得您想要的结果。