如何在PyTorch中初始化权重

Pytorch中loss.backward()和optimizer.step()的联系