pytorch
e26cb066 - squeeze: allow squeezing multiple dimensions at once (#89017)

Commit
1 year ago
squeeze: allow squeezing multiple dimensions at once (#89017) Ref #70924 This addresses part 1 of the issue, allowing `torch.squeeze` to be passed a tuple of dimensions. e.g. ```python x.squeeze(0).squeeze(0) ``` can now be written ```python x.squeeze((0, 1)) ``` (assuming x has at least 2 dimensions) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89017 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading