conv2d()、maxpool2d()、linear()、relu()等函数
基本函数torch.nn.Conv2d()torch.nn.MaxPool2d()torch.nn.Linear()torch.nn.ReLU()torch.nn.Conv2d()对由多个输入平面组成的输入信号应用 2D 卷积。class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')¶
复制链接