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的Tornado框架中的协程异步实现原理
Apr 23 Python
利用python批量修改word文件名的方法示例
Oct 17 Python
pandas值替换方法
Jul 10 Python
python多进程控制学习小结
Oct 31 Python
对python xlrd读取datetime类型数据的方法详解
Dec 26 Python
PyCharm的设置方法和第一个Python程序的建立
Jan 16 Python
基于Python实现迪杰斯特拉和弗洛伊德算法
May 27 Python
Pycharm新建模板默认添加个人信息的实例
Jul 15 Python
Django 实现admin后台显示图片缩略图的例子
Jul 28 Python
Python CVXOPT模块安装及使用解析
Aug 01 Python
Python Numpy,mask图像的生成详解
Feb 19 Python
PyTorch dropout设置训练和测试模式的实现
May 27 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 替换模板变量实现步骤
2009/08/24 PHP
解析php中memcache的应用
2013/06/18 PHP
php中函数前加&符号的作用分解
2014/07/08 PHP
PHP访问数据库集群的方法小结
2016/03/14 PHP
浅谈mysql_query()函数的返回值问题
2016/09/05 PHP
Javascript 布尔型分析
2008/12/22 Javascript
ExtJS PropertyGrid中使用Combobox选择值问题
2010/06/13 Javascript
EXTjs4.0的store的findRecord的BUG演示代码
2013/06/08 Javascript
js实现两个值相加alert出来精确到指定位
2013/09/25 Javascript
jquery获取tagName再进行判断
2014/05/29 Javascript
使用angular写一个hello world
2015/01/23 Javascript
javascript转换静态图片,增加粒子动画效果
2015/05/28 Javascript
javascript跨域总结之window.name实现的跨域数据传输
2015/11/01 Javascript
jQuery绑定事件on()与弹窗的简要概述
2016/04/27 Javascript
AngularJs IE Compatibility 兼容老版本IE
2016/09/01 Javascript
JS中from 表单序列化提交的代码
2017/01/20 Javascript
canvas 绘制圆形时钟
2017/02/22 Javascript
基于vue2框架的机器人自动回复mini-project实例代码
2017/06/13 Javascript
浅谈JS对html标签的属性的干预以及对CSS样式表属性的干预
2017/06/25 Javascript
JavaScript中的return布尔值的用法和原理解析
2017/08/14 Javascript
微信小程序倒计时功能实现代码
2017/11/09 Javascript
150行Node.js实现的dns代理工具
2019/08/02 Javascript
Vue过渡效果之CSS过渡详解(结合transition,animation,animate.css)
2020/02/05 Javascript
详解Vue3 Composition API中的提取和重用逻辑
2020/04/29 Javascript
three.js 制作动态二维码的示例代码
2020/07/31 Javascript
python如何修改装饰器中参数
2018/03/20 Python
Python使用百度api做人脸对比的方法
2019/08/28 Python
使用python实现时间序列白噪声检验方式
2020/06/03 Python
迪卡侬波兰体育用品商店:Decathlon波兰
2020/03/31 全球购物
第一批党的群众路线教育实践活动工作总结
2014/03/03 职场文书
电子商务专业自荐信
2014/06/02 职场文书
写字楼租赁意向书
2014/07/30 职场文书
我的梦想演讲稿500字
2014/08/21 职场文书
MySQL非空约束(not null)案例讲解
2021/08/23 MySQL
Java无向树分析 实现最小高度树
2022/04/09 Javascript
Hive日期格式转换方法总结
2022/06/25 数据库