pytorch的python API略读--torch.nn.Module

作者：机器视觉全栈er
网站：www.cvtutorials.com

2.3 torch.nn

本节我们看下torch.nn，这个模块是基础模块。

2.3.1 容器

torch.nn.Module：这个类是所有网络模块的基类，甚至你写普通的数学函数的时候，也可以用这个作为基类。内置的模块和你自定义的模块都可以相互嵌套，构成一个树状的模块结构。下面我们来看下这个Module类常用案例：

import torch
import torch.nn as nn
import torch.nn.functional as F

class MyModule(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(1, 3, 3)
        self.conv2 = nn.Conv2d(3, 3