Rearrange 函数
rearrange是einops中的一个函数调用方法
from einops import rearrange
具体使用方法
1.输入为图片
image = rearrange(image, 'h w c -> w h c') # 转置，对角线对称
2.输入为视频
image = rearrange(images, 'f h w c -> (f h) w c') # 在h维度合并
#(6, 300, 400, 3)
#(1800, 400, 3)