pytorch forward两个参数实例


Posted in Python onJanuary 17, 2020

以channel Attention Block为例子

class CAB(nn.Module):
 
  def __init__(self, in_channels, out_channels):
    super(CAB, self).__init__()
    self.global_pooling = nn.AdaptiveAvgPool2d(output_size=1)
    self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=1, stride=1, padding=0)
    self.relu = nn.ReLU()
    self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=1, stride=1, padding=0)
    self.sigmod = nn.Sigmoid()
 
  def forward(self, x):
    x1, x2 = x # high, low
    x = torch.cat([x1,x2],dim=1)
    x = self.global_pooling(x)
    x = self.conv1(x)
    x = self.relu(x)
    x = self.conv2(x)
    x = self.sigmod(x)
    x2 = x * x2
    res = x2 + x1
    return res

以上这篇pytorch forward两个参数实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用python分析git log日志示例
Feb 27 Python
Python中Scrapy爬虫图片处理详解
Nov 29 Python
python实现感知器算法详解
Dec 19 Python
树莓派动作捕捉抓拍存储图像脚本
Jun 22 Python
基于Python实现船舶的MMSI的获取(推荐)
Oct 21 Python
简单了解django处理跨域请求最佳解决方案
Mar 25 Python
Python爬虫HTPP请求方法有哪些
Jun 03 Python
怎么解决pycharm license Acti的方法
Oct 28 Python
python读写数据读写csv文件(pandas用法)
Dec 14 Python
Alpine安装Python3依赖出现的问题及解决方法
Dec 25 Python
详解win10下pytorch-gpu安装以及CUDA详细安装过程
Jan 28 Python
python 统计代码耗时的几种方法分享
Apr 02 Python
Python实现CNN的多通道输入实例
Jan 17 #Python
Python面向对象编程基础实例分析
Jan 17 #Python
通过python实现windows桌面截图代码实例
Jan 17 #Python
PyTorch加载预训练模型实例(pretrained)
Jan 17 #Python
python 正则表达式参数替换实例详解
Jan 17 #Python
Python函数式编程实例详解
Jan 17 #Python
python实现tail -f 功能
Jan 17 #Python
You might like
PHP pathinfo()获得文件的路径、名称等信息说明
2011/09/13 PHP
thinkphp实现发送邮件密码找回功能实例
2014/12/01 PHP
PHP+jQuery翻板抽奖功能实现
2015/10/19 PHP
PHP类的声明与实例化及构造方法与析构方法详解
2016/01/26 PHP
Zend Framework框架实现类似Google搜索分页效果
2016/11/25 PHP
PHP抽象类和接口用法实例详解
2019/07/20 PHP
PHP封装请求类实例分析【基于Yii框架】
2019/10/17 PHP
PHP全局使用Laravel辅助函数dd
2019/12/26 PHP
随鼠标上下滚动的jquery代码
2013/12/05 Javascript
js动态删除div元素基本思路及实现代码
2014/05/08 Javascript
IE6-IE9中tbody的innerHTML不能赋值的解决方法
2014/06/05 Javascript
Nodejs的express使用教程
2015/11/23 NodeJs
JavaScript禁止复制与粘贴的实现代码
2016/05/16 Javascript
深入解析Javascript闭包的功能及实现方法
2016/07/10 Javascript
jQuery自适应轮播图插件Swiper用法示例
2016/08/24 Javascript
详解javascript获取url信息的常见方法
2016/12/19 Javascript
使用JavaScript中的lodash编写双色球效果
2018/06/24 Javascript
JavaScript实现打砖块游戏
2020/02/25 Javascript
[42:24]完美世界DOTA2联赛循环赛 LBZS vs DM BO2第一场 11.01
2020/11/02 DOTA
[01:48:04]DOTA2-DPC中国联赛 正赛 PSG.LGD vs Elephant BO3 第一场 2月7日
2021/03/11 DOTA
python基础while循环及if判断的实例讲解
2017/08/25 Python
Tornado高并发处理方法实例代码
2018/01/15 Python
Windows上使用Python增加或删除权限的方法
2018/04/24 Python
python数据挖掘需要学的内容
2019/06/23 Python
flask框架渲染Jinja模板与传入模板变量操作详解
2020/01/25 Python
python使用docx模块读写docx文件的方法与docx模块常用方法详解
2020/02/17 Python
请编写一个 C 函数,该函数在给定的内存区域搜索给定的字符,并返回该字符所在位置索引值
2014/09/15 面试题
2019史上最全Database工程师题库
2015/12/06 面试题
建筑工程专业学生的自我评价
2013/12/25 职场文书
读书活动实施方案
2014/03/10 职场文书
会计试用期自我评价
2014/09/19 职场文书
领导班子在批评与自我批评座谈会上的发言
2014/09/28 职场文书
施工员岗位职责范本
2015/04/11 职场文书
自己搭建resnet18网络并加载torchvision自带权重的操作
2021/05/13 Python
HTML5页面音频自动播放的实现方式
2021/06/21 HTML / CSS
go语言使用Casbin实现角色的权限控制
2021/06/26 Golang