torch.linalg.diagonal

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

与torch.diagonal()具有默认值dim1=-2和dim2=-1的别名。