rearrange

einops.rearrange is a reader-friendly smart element reordering for multidimensional tensors. This operation includes functionality of transpose (axes permutation), reshape (view), squeeze, unsqueeze, stack, concatenate and other operations.

代替reshape,给维度命名。可以用…代表不想动的维度。 image.png image.png

使用 Hugo 构建
主题 Stack 由 Jimmy 设计