Loss.backward() raises error ‘grad can be implicitly created only for scalar outputs’

发表于 | 阅读次数:

输出的loss需要是一个值
loss.sum().backward()

请作者喝一杯咖啡☕️