标签： python tensorflow deep-learning theano
如何找到Theano的dimshuffle函数的Tensorflow等价物。
mask = mask.dimshuffle(0, 1, "x")
我这样做我尝试过很多方法却找不到。谢谢:)）
答案 0 :(得分：1)
您可以使用tf.transpose：
tf.transpose
mask = tf.transpose(mask, perm=[0, 1])