我想将一个大小为(n X N X M X m)的张量T转换为一个大小为(n X M X m)的张量U,同时只检索(NxN)块(即Uikl=Tiikl)上T的对角元素。torch.diag()只适用于2-D张量,我真的看不出如何在不对元素的索引进行循环的情况下做到这一点(考虑到我认为它的计算效率很低,我希望避免这种情况)。简而言之,我想对以下代码进行</em
) return i+1,l
我想以类似于上面代码描述的方式处理张量数组在while循环的主体中,我希望逐个元素地处理数组,以应用一些函数。为了进行演示,我给出了一个小代码片段。但是,它提供了如下错误消息。ValueError: Number of inputs and outputs of body mus