self.conv1 = torch.nn.Conv2d(input_planes,planes,stride=stride,kernel_size=3,padding=1
时间: 2023-04-06 13:01:42 浏览: 133
pytorch1.0中torch.nn.Conv2d用法详解
这是一个关于 PyTorch 的问题,self.conv1 是一个卷积层,input_planes 是输入通道数,planes 是输出通道数,stride 是步长,kernel_size 是卷积核大小,padding 是填充大小。
阅读全文