关于r1分钟理解pytorch的reshape函数中-1表示的意义_嘟教授的博客-CSDN博客_pytorch reshape
关于reshape的用法可以参考上面
X.reshape((-1, W.shape[0]))
关于autograd用法可参考
with torch.no_grad() 详解_失之毫厘,差之千里的博客-CSDN博客_with torch.no_grad():
pytorch中with torch.no_grad():_星之所望的博客-CSDN博客_with torch.no_grad()