文章归档

Pytorch中tensor的维度合并

本文介绍了在PyTorch中使用`view`函数合并tensor维度时保持数据顺序的方法。`view`通过将张量展平后按指定形状重组，但仅当合并连续维度时才能维持原始数据的分类意义（如batch维度）。若需合并非连续维度（例如将`(3,2,4,64)`变为`(3,128,4)`），直接使用`view`会导致数据错乱。此时应先用`permute`调整维度顺序（如变为`(3,2,64,4)`），再使用`view`合并，即可在重组后保持数据顺序与分类逻辑不变。