用einops包替换掉PyTorch的大部分矩阵变换函数
只是
rearrange ,reduce , repeat 和 einsum 这四个函数而已。 但真的很好用。 爱因斯坦求和标记:Einops tutorial, part 1: basics - Einops rearrange 函数:以更可读的方式重排、合并或拆分维度。 在这种计算机视觉的上下文里还挺合适的,分别是batch channel height…