pytorch
ef9f56eb - [primTorch] slice and transpose & etc.

Commit
2 years ago
[primTorch] slice and transpose & etc. This PR... Adds the following prims: - slice - slice_in_dim - transpose Adds the following refs: - cat - permute - transpose - swap_axes (alias for transpose) - tensor_split Makes the following test improvements: - adds reference inputs for torch.permute - adds a NumPy reference for torch.permute - adds reference inputs for torch.cat Fixes the following bugs: - adds support for scalars to the min and max prims Pull Request resolved: https://github.com/pytorch/pytorch/pull/76727 Approved by: https://github.com/ngimel
Author
Mike Ruberry
Committer
Parents
Loading