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)