美文网首页
初触PyTorch

初触PyTorch

作者: JalorOo | 来源:发表于2021-07-14 16:19 被阅读0次

    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

          本文链接:https://www.haomeiwen.com/subject/lebypltx.html