pytorch forward两个参数实例
更新时间：2020年01月17日 09:32:18 作者：weixin_41950276
今天小编就为大家分享一篇pytorch forward两个参数实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
以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列表推导式、嵌套、循环等相关操作技巧,需要的朋友可以参考下2019-10-10
这篇文章主要给大家介绍了关于如何用Python画五彩气球的相关资料,主要是用turtle库自带的画笔turtle.Turtle()来绘制气球,文中给出了详细的实例代码,需要的朋友可以参考下2023-06-06
一个完整的神经网络一般由三层构成：输入层，隐藏层（可以有多层）和输出层。本文所构建的神经网络隐藏层只有一层。一个神经网络主要由三部分构成（代码结构上）：初始化，训练，和预测。,需要的朋友可以参考下面文章内容的具体内容2021-09-09
这篇文章主要介绍了一些在 Python 编程中可能被忽视的核心功能，包括默认参数、海象运算符、*args 和 **kwargs 的使用等，需要的可参考下2023-08-08
这篇文章主要介绍了python uuid生成唯一id或str的最简单案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-01-01
在安装Python的Numpy包时,可能会遇到路径警告或包源超时的问题,首先,如果出现包源超时,可以尝试更换为国内的镜像源,如清华大学镜像源,其次,如果在安装完成后提示将某个路径添加到PATH环境变量,按照提示操作即可消除异常,需要的朋友可以参考下2024-09-09
这篇文章主要介绍了python中leastsq函数的使用方法，leastsq作用是最小化一组方程的平方和，下面文章举例说明详细内容，具有一的参考价值，需要的小伙伴可以参考一下2022-03-03
在 Python 中使用 wxPython 导出实体类列表数据到 Excel,通常可以借助 openpyxl 或 pandas 库来实现,下面就跟随小编一起来了解下具体操作吧2024-12-12
这篇文章主要介绍了Python图算法,结合实例形式详细分析了Python数据结构与算法中的图算法实现技巧,需要的朋友可以参考下2016-08-08
今天小编就为大家分享一篇利用pandas将非数值数据转换成数值的方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
最新评论