pytorch
cb23976f - Allow 0-dim batch sizes for AdaptiveMaxPool and MaxPool. (#62088)

Commit
3 years ago
Allow 0-dim batch sizes for AdaptiveMaxPool and MaxPool. (#62088) 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 `MaxPool` and `AdaptiveMaxPool` to accept tensors whose batch size is 0. Some changes have been made to modernize the tests so that they will show the name of C++ function that throws an error. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62088 Reviewed By: bdhirsh Differential Revision: D30281285 Pulled By: jbschlosser fbshipit-source-id: 52bffc67bfe45a78e11e4706b62cce1469eba1b9
Author
Parents
Loading