torch.linalg.diagonal

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

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