torch.triu_indices

row(int) -二维矩阵中的行数。

col(int) -二维矩阵中的列数。

offset(int) -与主对角线的对角线偏移。默认值：如果未提供，则为 0。

在 2-by-N 张量中返回矩阵的上三角部分（对角线上和之上的元素）的索引，其中第一行包含所有索引的行坐标，第二行包含列坐标。索引先按行排序，然后按列排序。

主对角线是i∈[0,min{d1,d2}−1] 的索引集{(i,i)}，其中d1,d2 是矩阵的维度。

（4,3,-1）沿主对角线向下偏移一位,相当于只要主对角线上方元素

（4,3,1）沿主对角线向上偏移一位，只要主对角线下方元素