视频地址土堆说卷积操作(可选看)_哔哩哔哩_bilibili
跟神经网路相关的工具都放在torch.nn模块里面torch.nn — PyTorch 2.5 documentation
nn.Conv1d | 在由几个输入平面组成的输入信号上应用一维卷积。 |
nn.Conv2d | 在由几个输入平面组成的输入信号上应用二维卷积。 |
nn.Conv3d | 在由几个输入平面组成的输入信号上应用三维卷积。 |
nn.ConvTranspose1d | 在由多个输入平面组成的输入图像上应用一维转置卷积算子。 |
nn.ConvTranspose2d | 在由多个输入平面组成的输入图像上应用二维转置卷积算子。 |
nn.ConvTranspose3d | 在由多个输入平面组成的输入图像上应用3D转置卷积算子。 |
import torch
import torch.nn.functional