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)