pytorch
4b52babc - [ONNX] Fix `any` and `all` outputs' shape (#79371)

Commit
2 years ago
[ONNX] Fix `any` and `all` outputs' shape (#79371) Part of #79263 Before: When `dim` == `None` and `keepdim` == `0`(`False`), the reduced output has `[1]` shape. After: Squeeze the output so that the shape will be `[]` as PyTorch's behavior. Pull Request resolved: https://github.com/pytorch/pytorch/pull/79371 Approved by: https://github.com/AllenTiTaiWang, https://github.com/BowenBao
Author
Committer
Parents
Loading