张量

本文介绍在pytorch中通过切片语法（如 [:, ::n, :]）高效提取高维张量第二维（索引为1）上每隔n个元素的方法，避免循环或复制，时间复杂度o(1)，适用于大规模张量处理…

本文详解如何使用 torch.gather 对形状为 [b, m, n] 的张量 A，按形状为 [b, k] 的索引张量 B 进行批量二维索引，得到 [b, k, n] 的输出张量…

本文详解如何使用 torch.gather 对 batched 3D 张量进行高效索引，通过维度扩展使索引张量与目标张量对齐，从而实现从 [b, m, n] 张量中按 [b, k]…