tensorflow中的矩阵转置

 import tensorflow as tf
 raw = tf.Variable(tf.random_normal(shape=(4,3,2)))#4个数组，3行两列
 a = tf.transpose(raw,perm=[2,0,1])#两个数组，4行3列
 b = tf.transpose(raw,perm=[1,0,2])#3个数组，4行2列
 with tf.Session() as sess:
     sess.run(tf.global_variables_initializer())
     print(raw.eval())
     print('**************************************')
     print(sess.run(a))
     print('**************************************')
     print(sess.run(b))