HWC 转换 CHW

a.shape()    #(100,100,3)

b=a.transpose(3, 1, 2)

b.shape()   #(3,100,100)