如何解决在派托奇寻找前k个比赛
我正在使用以下代码通过pytorch查找topk匹配项。
def find_top(self,x,y,n_neighbors,unit_vectors=False,cuda=False):
if not unit_vectors:
x = __to_unit_torch__(x,cuda=cuda)
y = __to_unit_torch__(y,cuda=cuda)
with torch.no_grad():
d = 1. - torch.matmul(x,y.transpose(0,1))
values,indices = torch.topk(d,dim=1,largest=False,sorted=True)
return indices.cpu().numpy()
不幸的是,它引发了以下错误
values,sorted=True)
RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THCTensorTopK.cu:23
d的大小为(1793,1)
。我想念什么?
解决方法
This error 当您使用大于类总数的 torch.topk
调用 k
时发生。减少你的争论,它应该运行良好。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。