grad can be implicitly created only for scalar outputs:修复 stack 后还要选择求导目标
-
PyTorch stack 后报 grad can be implicitly created only for scalar outputs 报错修复
这个报错的核心问题是,你用 torch.stack 拼接张量之后得到了多维…