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 23 Python
python清理子进程机制剖析
Nov 23 Python
对Python中数组的几种使用方法总结
Jun 28 Python
Pycharm之快速定位到某行快捷键的方法
Jan 20 Python
Python3.5面向对象程序设计之类的继承和多态详解
Apr 24 Python
python实现弹跳小球
May 13 Python
python获取磁盘号下盘符步骤详解
Jun 19 Python
python装饰器常见使用方法分析
Jun 26 Python
python读取图片的方式,以及将图片以三维数组的形式输出方法
Jul 03 Python
深入浅析python 中的self和cls的区别
Jun 20 Python
python interpolate插值实例
Jul 06 Python
python判断元素是否存在的实例方法
Sep 24 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
欧美媒体选出10年前最流行的17部动画
2017/01/18 日漫
MySQL中create table语句的基本语法是
2007/01/15 PHP
无法载入 mcrypt 扩展,请检查 PHP 配置终极解决方案
2011/07/18 PHP
PHP图片库imagemagick安装方法
2014/09/23 PHP
PHP计算日期相差天数实例分析
2016/02/23 PHP
PHP实现的注册,登录及查询用户资料功能API接口示例
2017/06/06 PHP
JavaScript 基础知识 被自己遗忘的
2009/10/15 Javascript
javascript setTimeout和setInterval 的区别
2009/12/08 Javascript
JavaScript页面刷新与弹出窗口问题的解决方法
2010/03/02 Javascript
js几个验证函数代码
2010/03/25 Javascript
JS不间断向上滚动效果代码
2013/12/25 Javascript
详解使用nvm管理多版本node的方法
2017/08/30 Javascript
jquery自定义显示消息数量
2017/12/19 jQuery
微信小程序移动拖拽视图-movable-view实例详解
2019/08/17 Javascript
Python装饰器使用实例:验证参数合法性
2015/06/24 Python
python 环境变量和import模块导入方法(详解)
2017/07/11 Python
Django 添加静态文件的两种实现方法(必看篇)
2017/07/14 Python
python如何统计序列中元素
2020/07/31 Python
python 给DataFrame增加index行名和columns列名的实现方法
2018/06/08 Python
python+numpy+matplotalib实现梯度下降法
2018/08/31 Python
Python类的继承、多态及获取对象信息操作详解
2019/02/28 Python
python中@property和property函数常见使用方法示例
2019/10/21 Python
JupyterNotebook设置Python环境的方法步骤
2019/12/03 Python
基于python 凸包问题的解决
2020/04/16 Python
Python爬虫之Selenium设置元素等待的方法
2020/12/04 Python
HTML5 Blob 实现文件下载功能的示例代码
2019/11/29 HTML / CSS
AmazeUI 图标的示例代码
2020/08/13 HTML / CSS
Myprotein蛋白粉美国官网:欧洲畅销运动营养品牌
2016/11/15 全球购物
维珍澳洲航空官网:Virgin Australia
2017/09/08 全球购物
FC-Moto瑞典:欧洲最大的摩托车服装和头盔商店之一
2018/11/27 全球购物
小米俄罗斯授权商店:Xiaomi俄罗斯
2019/12/08 全球购物
大学毕业通用个人的求职信
2013/12/08 职场文书
出生公证书样本
2014/04/04 职场文书
团结就是力量演讲稿
2014/05/21 职场文书
化学专业自荐信
2014/05/28 职场文书
实习报告怎么写
2019/06/20 职场文书