torch.Tensor.diagonal

Tensor.diagonal(offset=0, dim1=0, dim2=1) → Tensor

参见 torch.diagonal()