einops.rearrange、repeat、reduce 对数据维度进行操作