torch.linalg.diagonal

torch.linalg.diagonal(A, *, offset=0, dim1=- 2, dim2=- 1) → Tensor

Alias for torch.diagonal() with defaults dim1= -2, dim2= -1.