torch.nn.functional.conv2d 函数详解

形式:torch.nn.functional.conv2d(input,filters,bias,stride,padding,dilation,groups)

返回值:一个Tensor变量

作用:在输入图像input中使用filters做卷积运算

参数的具体意义:

input代表输入图像的大小(minibatch,in_channels,H,W),是一个四维tensor

filters代表卷积核的大小(out_channels,in_channe/groups,H,W),是一个四维tensor

bias代表每一个channel的bias,是一个维数等于out_channels的tensor

stride是一个数或者一个二元组(SH,SW),代表纵向和横向的步长

padding是一个数或者一个二元组(PH,PW ),代表纵向和横向的填充值

dilation是一个数,代表卷积核内部每个元素之间间隔元素的数目(不常用,默认为0,可以不用管,如果想详细了解,请参考:https://www.cnblogs.com/wanghui-garcia/p/10775367.html

groups是一个数,代表分组卷积时分的组数,特别的当groups = in_channel时,就是在做逐层卷积(depth-wise conv)

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值