torch是一个类, tensor是其下的一个多维矩阵类的封装。
requires_grad = True
requires_grad 标记
requires_grad标记记录以后计算的行踪,之后进行back_prop时进行反向传播求导。值得注意的是,只会对叶子节点求导,不会对中间节点求导。
class torch.nn.Module
torch.nn.Module是所有网络的基类。我们创建的任何模型都应该继承这个类。
import torch.nn as nn
import torch.nn.functional as F
class Model(nn.Module):
def