Q1:对Pytorch中backward()函数的理解
A: 如果需要计算导数,可以在Tensor上调用.backward()。
- 如果Tensor是一个标量(即它包含一个元素的数据),则不需要为backward()指定任何参数。
- 但是如果它有更多的元素,则需要指定一个gradient参数,它是形状匹配的张量。
出处:https://blog.csdn.net/weixin_43763731/article/details/88982979
Q2:对Pytorch中nn.Module
中重载forward()
函数理解
A:
model = LeNet()
y = model(x)
如上则调用网络模型定义的forward方法
网友评论