tree 0a8d18db9c7f8613b34e0914e04141b7aa08f79b
parent fe277b87171a378e48798dceb938dffdebb56e81
author lezcano <lezcano-93@hotmail.com> 1643182789 -0800
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1643184512 +0000

[Array API] Add linalg.diagonal (#70599)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70599

This PR adds `linalg.diagonal` following the Array API:
https://data-apis.org/array-api/latest/extensions/linear_algebra_functions.html#linalg-diagonal-x-axis1-0-axis2-1-offset-0

Fixes https://github.com/pytorch/pytorch/issues/62813

cc jianyuh nikitaved pearu mruberry walterddr IvanYashchuk xwang233 Lezcano rgommers pmeier asmeurer leofang AnirudhDagar asi1024 emcastillo kmaehashi

Test Plan: Imported from OSS

Reviewed By: albanD

Differential Revision: D33760506

Pulled By: mruberry

fbshipit-source-id: e32c3490321d8c3f31b3bb538bc1f72b39bd2854
(cherry picked from commit 44f41f8e3922892ca2f86c9c05249336de40e9ee)