有没有办法可以在不使用循环的情况下多次应用torch.mode

问题描述

让我们说arr1的大小为10，arr2的大小为5乘5。（arr2包含5行要检查的目标索引） 有没有一种方法可以使用5个不同的索引获取arr1的模式而无需循环

我尝试过这种方法，但是只有在arr2是一维的情况下才有效，并且它返回arr2中特定索引的值的模式

modes =torch.mode(arr1[arr2]).values 

解决方法

我试过了

arr1 = torch.tensor([2,4,5,7,6,5])
arr2 = torch.tensor([[0,1,2,3,4],[5,8,9]])

arr1.shape
torch.Size([10])
arr2.shape
torch.Size([2,5])

torch.mode(arr1[arr2]).values
tensor([5,6])