-
Notifications
You must be signed in to change notification settings - Fork 27.7k
Support expand_dims #56774
Copy link
Copy link
Open
Labels
function requestA request for a new function or the addition of new arguments/modes to an existing function.A request for a new function or the addition of new arguments/modes to an existing function.module: numpyRelated to numpy support, and also numpy compatibility of our operatorsRelated to numpy support, and also numpy compatibility of our operatorsmodule: python array apiIssues related to the Python Array APIIssues related to the Python Array APItriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
Metadata
Metadata
Assignees
Labels
function requestA request for a new function or the addition of new arguments/modes to an existing function.A request for a new function or the addition of new arguments/modes to an existing function.module: numpyRelated to numpy support, and also numpy compatibility of our operatorsRelated to numpy support, and also numpy compatibility of our operatorsmodule: python array apiIssues related to the Python Array APIIssues related to the Python Array APItriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
🚀 Feature
Support
torch.expand_dims.(cc: @mruberry, @rgommers, @emcastillo and @kmaehashi)
Motivation and Pitch
To enhance the compatibility with NumPy’s interface (c.f. #38349).
Currently PyTorch supports the same functionality with a different function name:
torch.unsqueeze, but I would like to support an alias to improve compatibility with NumPy and Python array API standard (#54581).References
cc @mruberry @rgommers @heitorschueroff @pmeier