在 patch_trg 中,我无法理解为什么你会这样改变数据形状

作者: kwanhoP创建于 2023年3月8日更新于 2023年6月16日

我的数据集是以水平方式组成的,我没有使用 transpose(0,1),所以我将你的代码修改如下所示:

内容来源: jadore801120/attention-is-all-you-need-pytorch