Fix errors when initializing Linear with 0 in_features (#56505)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/48152
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56505
Reviewed By: malfet
Differential Revision: D27919590
Pulled By: jbschlosser
fbshipit-source-id: 462ca280051f63c31ff588c38a9e436116c0f336