得到多个维度上的对角线元素我想把一个大小为(n x n x m x m)的张量T转换成一个大小为(n x m x m)的张量U,同时只检索(NxN)块上T的对角线元素(即Uikl=Tiikl)。diag()只适用于二维张量,我 ...2024-09-30 已阅读: n次