Pytorch_nn.Conv2d

本文详细介绍了PyTorch中Conv2d函数的参数及其作用,包括输入和输出通道数、卷积核大小、步长、填充、膨胀率、组数等,深入解析了这些参数如何影响卷积层的工作原理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Conv2d

torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, 
    padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')
  • in_channels 输入数据通道数
  • out_channels 输出数据通道数
  • kennel_size 卷积核大小,int或tuple
  • stride 步长
  • padding 每个维度零填充的数量
  • dilation 内核点之间的距离,也称à trous algorithm
  • groups 控制inputs与outputs间的连接
  • groups=1,所有输入都卷积到输出
  • groups=2,并排设置两个conv层,每个层查看一半的输入通道,并生成一半的输出通道,然后将两者连接起来
  • groups=in_channels,每个输入通道都有它自己的filter,size为[out_channels/in_channels]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值