我如何找到与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/