初触PyTorch

Q1:对Pytorch中backward（）函数的理解

A: 如果需要计算导数，可以在Tensor上调用.backward()。

  1. 如果Tensor是一个标量（即它包含一个元素的数据），则不需要为backward()指定任何参数。
  2. 但是如果它有更多的元素，则需要指定一个gradient参数，它是形状匹配的张量。
    出处：https://blog.csdn.net/weixin_43763731/article/details/88982979

Q2：对Pytorch中nn.Module中重载forward()函数理解

A：

model = LeNet()
y = model(x)

如上则调用网络模型定义的forward方法

出处：(https://blog.csdn.net/u011501388/article/details/84062483)[https://blog.csdn.net/u011501388/article/details/84062483]

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容