Tensorflow相当于Theano的dimshuffle

Question

如何找到与 Theano 的 dimshuffle 函数等效的 Tensorflow。

mask = mask.dimshuffle(0, 1, "x")

我正在做这个我试了很多方法都找不到。谢谢:)

Answer 1

您可以使用 tf.transpose:

 mask = tf.transpose(mask, perm=[0, 1])