Pytorch：快速求得NxN矩阵的主对角线(diagonal)元素与非对角线元素_我是大黄同学呀的博客_pytorch 取对角线

前因

今天写代码过程中，要分别求得一个N x NTensor矩阵的主对角元素与非对角元素，看到一个巧妙的写法，故记录下来。Tensor矩阵的主对角元素