pytorch
36ceaeeb - Allow LocalResponseNorm to accept 0 dim batch sizes (#62801)

Commit
3 years ago
Allow LocalResponseNorm to accept 0 dim batch sizes (#62801) Summary: This issue fixes a part of https://github.com/pytorch/pytorch/issues/12013, which is summarized concretely in https://github.com/pytorch/pytorch/issues/38115. This PR allows `LocalResponseNorm` to accept tensors with 0 dimensional batch size. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62801 Reviewed By: zou3519 Differential Revision: D30165282 Pulled By: jbschlosser fbshipit-source-id: cce0b2d12dbf47dc8ed6247c267bf2f2305f858a
Author
Committer
Parents
Loading