-
Notifications
You must be signed in to change notification settings - Fork 27.7k
Support torch.linalg.diagonal #62813
Copy link
Copy link
Closed
Labels
module: 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
Metadata
Metadata
Assignees
Labels
module: 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.linalg.diagonalalias to torch.diagonal.cc @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi @heitorschueroff
Motivation & Pitch
Python Array API proposes the use of
linalg.diagonalamong other linear algebra functions. Currently, PyTorch supports the same functionality withtorch.diagonal. Array-API Ops tracker (#58742) lists a few aliases and it would be nice to add an aliastorch.linalg.diagonaltotorch.diagonalmaking it compliant with Array API.Additional context