pytorch
64c6a89b - [primTorch] reshape and view (#77220)

Commit
2 years ago
[primTorch] reshape and view (#77220) This PR makes the following changes... Prims - adds as_strided - fixes errors in flatten meta Testing - enables view consistency checking (which can be opted out of, see issues below) - adds reference inputs for view, reshape, and flatten - adds error inputs for reshape Refs - adds as_strided, reshape, and view - fixes an error in the flatten ref where it was not returning self on no-op - fixes a bug in transpose where it was not retuning a view when the transposed tensor has 1 or fewer dims Issues - https://github.com/pytorch/pytorch/issues/77218 - https://github.com/pytorch/pytorch/issues/77216 Pull Request resolved: https://github.com/pytorch/pytorch/pull/77220 Approved by: https://github.com/ngimel
Author
Mike Ruberry
Committer
Parents
Loading