简要回顾 PyTorch 中关于 Tensor 形状变换的 API(如 view, reshape, unsqueeze, squeeze)以及 Einops 库中 rearrange, reduce, repeat 的用法。
2024-09-16