-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Closed
Labels
module: linear algebraIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulmodule: 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
🚀 Feature
Support torch.linalg.cross.
cc @jianyuh @nikitaved @pearu @mruberry @heitorschueroff @walterddr @IvanYashchuk @xwang233 @lezcano @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi
Motivation & Pitch
Python Array API proposes the use of linalg.cross among other linear algebra functions. Currently, PyTorch supports the same functionality with torch.cross. Array-API Ops tracker (#58742) lists a few aliases and it would be nice to add an alias torch.linalg.cross to torch.cross making it compliant with Array API.
Additional context
rgommers, kmaehashi and mruberry
Metadata
Metadata
Assignees
Labels
module: linear algebraIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulmodule: 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