PyTorch Core 更新:修正無效卷積輸入形狀錯誤 (#201058)
trunk/895047b6cb93b5d9ffec57224caff08a9af1cff6: Reject invalid conv input shapes with clear errors (#201058)
重點導讀
PyTorch 在此更新中修正了 CUDNN 相關的尺寸檢查缺陷,避免了輸入形狀不符時的 IMAs 與錯誤訊息。
- 關鍵修正:新增對輸入維度 >= 3 的檢查,移動 checkDimRange 前置。
AI 伴隨問答
針對這篇報導有疑問?可直接點選快捷問題或向 AI 追問背景脈絡與技術細節:
信源出處github.com
trunk/895047b6cb93b5d9ffec57224caff08a9af1cff6: Reject invalid conv input shapes with clear errors (#201058)
本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。
來源: PyTorch Core Releases (github.com)
開啟原文