pytorch报错:Index_select(): functions with out=… arguments don’t support automatic differentiation
解决:
将注释的代码改为如下就不报错了,原因不清楚,有懂的大佬求指点
# torch.index_select(x1, 0, idx, out=xx1)
xx1 = torch.index_select(x1, 0, idx)
# torch.index_select(y1, 0, idx, out=yy1)
yy1 = torch.index_select(y1, 0, idx)
# torch.index_select(x2, 0, idx, out=xx2)
xx2 = torch.index_select(x2, 0, idx)
# torch.index_select(y2, 0, idx, out=yy2)
yy2 = torch.index_select(y2, 0, idx)
原因:感觉可能是api改了。
PyTorch自动微分错误解决方案
这篇博客讨论了PyTorch中遇到的'Index_select()函数不支持自动微分'的问题,提供了一种修改代码避免错误的方法。通过将使用'out='参数的调用替换为独立的赋值语句,解决了报错问题。原因可能是API更新导致的不兼容。
6228

被折叠的 条评论
为什么被折叠?



