-
Notifications
You must be signed in to change notification settings - Fork 27.2k
Open
Labels
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 APImodule: reductionstriagedThis 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
The following reduction operators should be updated to allow reducing over multiple dimensions.
For the following operators, adding support for reducing over multiple dimensions is blocked on the following issue #61490
Related Issues
- Support multi-dim reductions for torch.prod, torch.all, torch.any #56586
- Have the possibility to reduce a tensor with median on more than one specified dimension #53900
- Reduce with any(), all(), median() over multiple dimensions #61453
cc @mruberry @rgommers @heitorschueroff @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
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 APImodule: reductionstriagedThis 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