pytorch
1c42b946 - [ONNX] Update support of exporting bool type index mask (#32445)

Commit
5 years ago
[ONNX] Update support of exporting bool type index mask (#32445) Summary: e.g. `tensor[torch.tensor([0, 1, 0], dtype=torch.bool)]` Previously the mask is of type uint8. Both uint8 and bool should be supported for export. Pull Request resolved: https://github.com/pytorch/pytorch/pull/32445 Reviewed By: hl475 Differential Revision: D19610713 Pulled By: houseroad fbshipit-source-id: 8df636e0c3cb0b82919a689242a962c79220209c
Author
Parents
Loading