torch.Conv2d
时间: 2023-11-23 20:47:59 浏览: 111
pytorch1.0中torch.nn.Conv2d用法详解
torch.nn.Conv2d()是PyTorch中的一个函数,用于创建卷积神经网络中的卷积层。它有多个参数,包括in_channels(输入图像的通道数),out_channels(输出图像的通道数),kernel_size(卷积核的大小),stride(步长),padding(边缘填充),dilation(扩张因子),groups(分组卷积),bias(是否使用偏置项)和padding_mode(填充模式)。该函数的具体用法和tensorflow中的类似,但并不完全相同。
阅读全文