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中的序列化与反序列化的使用
Jun 30 Python
python实现识别相似图片小结
Feb 22 Python
Python中对象迭代与反迭代的技巧总结
Sep 17 Python
解决PyCharm中光标变粗的问题
Aug 05 Python
浅谈Python peewee 使用经验
Oct 20 Python
python-tkinter之按钮的使用,开关方法
Jun 11 Python
对python中不同模块(函数、类、变量)的调用详解
Jul 16 Python
Python通过文本和图片生成词云图
May 21 Python
用python对excel查重
Dec 07 Python
Python多线程 Queue 模块常见用法
Jul 04 Python
python+opencv实现目标跟踪过程
Jun 21 Python
python语言中pandas字符串分割str.split()函数
Aug 05 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下使用以下代码连接并测试
2008/04/09 PHP
php中文验证码实现示例分享
2014/01/12 PHP
php使用ob_start()实现图片存入变量的方法
2014/11/14 PHP
php中通过DirectoryIterator删除整个目录的方法
2015/03/13 PHP
Laravel框架使用Seeder实现自动填充数据功能
2018/06/13 PHP
利用PHP扩展Xhprof分析项目性能实践教程
2018/09/05 PHP
PHP数组基本用法与知识点总结
2020/06/02 PHP
javascript 清空form表单中某种元素的值
2009/12/26 Javascript
使用Math.floor与Math.random取随机整数的方法详解
2013/05/07 Javascript
js实现的点击数量加一可操作数据库
2014/05/09 Javascript
JavaScript中的方法重载实例
2015/03/16 Javascript
javascript合并表格单元格实例代码
2016/01/03 Javascript
字符串反转_JavaScript
2016/04/28 Javascript
AngularJS实现数据列表的增加、删除和上移下移等功能实例
2016/09/05 Javascript
angular.js 路由及页面传参示例
2017/02/24 Javascript
zTree 树插件实现全国五级地区点击后加载的示例
2018/02/05 Javascript
js与jQuery实现获取table中的数据并拼成json字符串操作示例
2018/07/12 jQuery
vue+axios+mock.js环境搭建的方法步骤
2018/08/28 Javascript
angularJs中$scope数据序列化的实例
2018/09/30 Javascript
layui点击弹框页面 表单请求的方法
2019/09/21 Javascript
JavaScript实现网页计算器功能
2020/10/29 Javascript
Python中强大的命令行库click入门教程
2016/12/26 Python
Python Flask基础教程示例代码
2018/02/07 Python
Python中pow()和math.pow()函数用法示例
2018/02/11 Python
详解python中@的用法
2019/03/27 Python
python实现两个dict合并与计算操作示例
2019/07/01 Python
财务管理专业毕业生求职信范文
2013/09/21 职场文书
英语专业大学生求职简历的自我评价
2013/10/18 职场文书
广告传媒专业应届生求职信
2014/03/01 职场文书
报关员个人职业生涯规划书
2014/03/12 职场文书
超级礼物观后感
2015/06/15 职场文书
厉行节约工作总结
2015/08/12 职场文书
如何用threejs实现实时多边形折射
2021/05/07 Javascript
Vue和Flask通信的实现
2021/05/19 Vue.js
Python实现自动玩连连看的脚本分享
2022/04/04 Python
Java工作中实用的代码优化技巧分享
2022/04/21 Java/Android