PyTorch第一步--01基础定义
nn.Module是nn中最重要的类,可把它看成是一个网络的封装,包含网络各层定义以及forward方法,调用forward(input)方法,在`forward` 函数中可使用任何tensor支持的函数,还可以使用if、for循环、print、log等Python语法,写法和标准的Python写法一致。除了这些预定义的层和函数外,torch.nn还提供了Module类,该类作为一个基类,允许用户自定义自己的层和模型,并重写forward()函数实现前向传播过程。** 的函数会修改Tensor本身。
复制链接