如何在TensorFlow中置入光明？

Question

我能够缩小张量的偏移：我可以在Pytorch中做到这一点！但是不在张量之类的地方！

A = torch.rand(1, 2,5)
A = A.permute(0,2,1) 
A.shape

Torch.Size（[1，5，2]）

TensorFlow（只是尝试一下，我不知道）：

A = tf.random.normal(1, 2,5)
A = tf.keras.layers.Permute((0, 2, 1))

不工作

Answer 1

使用TF.Transpose：

import tensorflow as tf

A = tf.random.normal((1, 2, 5))

A_t = tf.transpose(A, perm=[0, 2, 1])

print(A.shape, A_t.shape)
# (1, 2, 5) (1, 5, 2)