torch.linalg.diagonal

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

别名，等同于 torch.diagonal()，默认 dim1= -2, dim2= -1。