报错如下:
原因: 这里的loss默认的requires_grad是False,因此在backward()处不会计算梯度,导致出错
方法:将loss的requires_grad属性设为True,如下所示: