-
Notifications
You must be signed in to change notification settings - Fork 27.7k
Support torch.linalg.trace #62255
Copy link
Copy link
Open
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
Description
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.trace.In addition to the features of
torch.trace,axis1,axis2andoffsetoptions are also wanted.(cc: @mruberry, @rgommers, @emcastillo and @kmaehashi)
Motivation and Pitch
To enhance the compatibility with NumPy’s interface and Python array API standard.
Needs discussions
Should the axis name be
axis1andaxis2, ordim0anddim1? Both should be supported?References
cc @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi @heitorschueroff