RuntimeError: invalid argument 4: Input tensor must have same dimensions as

系统管理员 2023-06-27 03:41 69阅读 0赞

在使用scatter_的时候,出现了这个问题

这个问题纠结了挺久,后来发现是自己给自己挖的坑,因为我们可以看一下scatter_这个函数

scatter_(int dim, Tensor index, Tensor src)

这个方法要求我们第三个参数为tensor,然后我将数字改成了Tensor就出现了 这个错

原来是自己很蠢,其实还有一个形式

scatter_(int dim, Tensor index, Number value)

也就是第三个可以是数字的,不一定得是Tensor呢

然鹅,搜网上的资料基本上都是说修改维度,关键最难受的是我不知道我需要维度修改成啥样

毕竟自己这么蠢的人还是,这么低级的错误

发表评论

表情:
评论列表 (有 0 条评论,69人围观)

还没有评论,来说两句吧...

相关阅读