pytorch
f6c7081a - Allow FractionalMaxPool 2D and 3D layers to accept 0 dim batch size tensors. (#62083)

Commit
3 years ago
Allow FractionalMaxPool 2D and 3D layers to accept 0 dim batch size tensors. (#62083) 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. Allow `FractionalMaxPool` 2D and 3D layers to accept 0 dim batch sizes. Also make some minor corrections to error messages to make them more informative. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62083 Reviewed By: H-Huang Differential Revision: D30134461 Pulled By: jbschlosser fbshipit-source-id: 0ec50875d36c2083a7f06d9ca6a110fb3ec4f2e2
Author
Parents
Loading