hanxuwerewolf的博客

import torch from torch import nn import torch.nn.functional as F # 首先建立一个全连接的子module，继承nn.Module class Linear(nn.Module): def __init__(self, in_dim, out_dim): super(Linear, self).__init__() # 调...

首先需要知道梯度，梯度下降等相关概念，这里传送门：https://www.jianshu.com/p/17191c57d7e9 import torch def test(): """ requires_grad表示是否对该tensor进行求导，默认为False; 设置为True则需要求导，并且依赖于该Tensor的之后的所有节点都需要求导。 其中： grad:该tenso...