Fix #92814: assertion error when explicitly provide out=None (#92873)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/92873
Approved by: https://github.com/albanD, https://github.com/bdhirsh
