pytorch forward两个参数实例
作者：weixin_41950276 发布时间：2022-09-05 09:54:34
标签：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
来源：https://blog.csdn.net/weixin_41950276/article/details/89069659
0
投稿猜你喜欢
- 在web.config文件中添加<connectionStrings><add name="SQLConnect
在transactional replication, 经常会遇到数据同步延迟的情况。有时候这些延迟是由于在publication中执行了一- 一个封装好的JavaScript拖动类，使用方便：<div id="idDrag" style="bor
- 1. 矩阵求逆import numpy as npa = np.array([[1, 2], [3, 4]]) # 初始化一个非奇异矩阵(数
- 下面的例子演法了怎么样从协程里返回一个值：import asyncioasync def coroutine(): print(
- 本文实例讲述了python实现通过代理服务器访问远程url的方法。分享给大家供大家参考。具体如下：import urllibproxies
- 框架介绍在之前的.NET中，微软还没有提供过像样的日志框架，目前能用的一些框架比如Log4Net、NLog、CommonLogging使用起
- 前言Python 有 while 语句和 for 语句作为循环处理。虽然 for 语句具有一定数量的进程，但 while 语句是『直到满足条
- 调用：var pageChange = function (index) { &n
- PyQt5布局控件QVBoxLayout简介采用QVBoxLayout类，按照从上到下的顺序添加控件本节内容较少，演示两个实例，便于明白QV
- 目前代码应该没什么bug了，兼容IE6.0 & FF 1.5, 通过xHTML 的Transitional验证和 CSS 验证。为了
- ??本文核心：setup()概念、 reactive()的使用【前言】vue3作为vue2的升级版，有着很多的新特性，其中就包括了组合式AP
- 要编写一个简单的能发布网页的 Python 应用服务器，可以使用 Python 自带的 http.server 模块来实现。http.ser
- 检测自己当前系统环境中python是否已经安装该module，若未安装请自行安装检测自己的pycharm使用的环境变量是否与当前环境一致若不
- 前言在学校太闲，就写了这个程序，可以绘制函数图像，并且可以绘制其导函数图像和不定积分的图像，效果非常不错。效果图说明1，程序无法绘制复数图像
- 大多数卷积神经网络都是直接通过写一个Model类来定义的，这样写的代码其实是比较好懂的，特别是在魔改网络的时候也很方便。然后也有一些会通过c
- 没什么实际用途，纯属消遣Quick Click<html><head><title>Quick_Clic
- 前言不管是做开发还是做过网站的朋友们，应该对于User Agent一点都不陌生，User Agent 中文名为用户代理，简称 UA，它是一个
- np.percentilenumpy.percentile(a, q, axis=None, out=None, overwrite_inp
- sql xml 入门: --by jinjazz 1、xml: 能认识元素、属性和值 2、xpath: 寻址语言，类似windows目录的查
