tf.transpose(input, [dimension_1, dimenaion_2,…,dimension_n]):
这个函数主要适用于交换输入张量的不同维度用的
import tensorflow as tf;
import numpy as np;
A = np.array([[1,2,3],[4,5,6]])
x = tf.transpose(A, [1,0])
B = np.array([[[1,2,3],[4,5,6]]])
y = tf.transpose(B, [2,1,0])