PyTorch 2.14.1-rc2 修复 MPS 上 Jacobi SVD 收敛问题
PyTorch 发布 v2.14.1-rc2,修复 MPS 后端 Jacobi SVD 在列数较少时的收敛缺陷。原实现中,当两列范数乘积低于 float32 epsilon 时会跳过旋转,即使两列高度相关,导致病态输入下 U 矩阵非正交、小奇异值被放大。新代码对小的非零列改用相对相关性判据,并保留零空间补全。测试在 test_linalg.py 中新增 float32、float64、complex64、complex128 的正交性、重构、奇异值和数值秩覆盖,不使用固定随机种子。该 PR 由 Codex 协助编写…
推荐理由在 Mac 上用 MPS 跑 SVD 或依赖它的降维、推荐算法的开发者,升级后小奇异值结果更可靠。