pytorch
4615f6aa - [MPS]: Add fix for squeezed input axes handling in BCE loss (#79676)

Commit
3 years ago
[MPS]: Add fix for squeezed input axes handling in BCE loss (#79676) Fixes #79527 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79676 Approved by: https://github.com/razarmehr, https://github.com/albanD
Author
Committer
Parents
Loading