torch.linalg.diagonal

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

torch.diagonal() 的别名，默认值 dim1 = -2， dim2 = -1。