pytorch
cfac765a - Allow Average Pooling modules to accept tensors with 0-dim batch sizes. (#62025)

Commit
3 years ago
Allow Average Pooling modules to accept tensors with 0-dim batch sizes. (#62025) 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. It introduces changes and tests for allowing the Average Pooling layers to accept tensors with 0 sized batch dimensions and return meaningful results. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62025 Reviewed By: VitalyFedyunin Differential Revision: D30303256 Pulled By: jbschlosser fbshipit-source-id: 5f727e62a7c58d2b8bb49fcc3bd7688474917ba5
Author
Committer
Parents
Loading