tf.transpose简单说明

三维张量:

x = [[[ 1  2  3  4]
  [ 5  6  7  8]
  [ 9 10 11 12]]

 [[21 22 23 24]
  [25 26 27 28]
  [29 30 31 32]]]

x是一个高2个3行4列的张量

tf.transpose(x,[0,1,2])是x张量的原型,其中0表示高为2,1表示行为3,2表示列为4

tf.transpose(x,[0,1,2])直观的表示为:x.reshape(2,3,4)

tf.transpose(x,[1,0,2])直观的表示为:x.reshape(3,2,4)

tf.transpose(x,[1,2,0])直观的表示为:x.reshape(3,4,2)

tf.transpose(x,[2,0,1])直观的表示为:x.reshape(4,2,3)

tf.transpose(x,[2,1,0])直观的表示为:x.reshape(4,3,2)