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

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


我正在这样做,我尝试了很多方法但找不到。谢谢 :)

最佳答案

您可以使用tf.transpose

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

关于python - Tensorflow相当于Theano的dimshuffle,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/39790829/

10-12 19:06