初触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]

你可能感兴趣的:(初触PyTorch)