pytorch 计算ConvTranspose1d输出特征大小方式

更新时间：2020年06月23日 11:10:32 作者：明月几时有.

这篇文章主要介绍了pytorch 计算ConvTranspose1d输出特征大小方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

问题：如何经过convTransposed1d输出指定大小的特征？

import torch
from torch import nn
import torch.nn.functional as F

conv1 = nn.Conv1d(1, 2, 3, padding=1)
conv2 = nn.Conv1d(in_channels=2, out_channels=4, kernel_size=3, padding=1)
#转置卷积
dconv1 = nn.ConvTranspose1d(4, 1, kernel_size=3, stride=2, padding=1, output_padding=1)

x = torch.randn(16, 1, 8)
print(x.size())

x1 = conv1(x)
x2 = conv2(x1)
print(x2.size())

x3 = dconv1(x2)
print(x3.size())

'''
torch.Size([16, 1, 8])
torch.Size([16, 4, 8]) #conv2输出特征图大小
torch.Size([16, 1, 16]) #转置卷积输出特征图大小
'''

#转置卷积
dconv1 = nn.ConvTranspose1d(1, 1, kernel_size=3, stride=3, padding=1, output_padding=1)

x = torch.randn(16, 1, 8)
print(x.size()) #torch.Size([16, 1, 23])

x3 = dconv1(x)
print(x3.size()) #torch.Size([16, 1, 23])

下面两图为演示conv1d，在padding和不padding下的输出特征图大小

不带padding

带padding

补充知识：判断pytorch是否支持GPU加速

如下所示：

print torch.cuda.is_available()

以上这篇pytorch 计算ConvTranspose1d输出特征大小方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python循环语句介绍
大家好，本篇文章主要讲的是Python循环语句介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览
2021-12-12
Python 的第三方调试库 pysnooper 使用示例
这篇文章主要介绍了Python 的第三方调试库 pysnooper 使用示例的相关资料,需要的朋友可以参考下
2023-02-02
python调用系统ffmpeg实现视频截图、http发送
这篇文章主要为大家详细介绍了python调用系统ffmpeg实现视频截图、http发送，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-03-03
在多种情况/开发环境中运行python脚本和代码的技巧分享
Python脚本或程序是包含可执行Python代码的文件,能够运行Python脚本和代码可能是您作为Python开发人员所需的最重要的技能,在本教程中,您将学习一些运行Python脚本和代码的技术,在每种情况下使用的技术将取决于您的环境、平台、需求和技能
2023-11-11
python自动化之Ansible的安装教程
这篇文章主要介绍了python自动化之Ansible的安装方法,结合实例形式分析了自动化运维工具Ansible的安装步骤及相关操作命令,需要的朋友可以参考下
2019-06-06
详解python算法常用技巧与内置库
这篇文章主要介绍了python算法常用技巧与内置库,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-10-10
python对指定字符串逆序的6种方法(小结)
这篇文章主要介绍了python对指定字符串逆序的6种方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-04-04
Python处理PDF及生成多层PDF实例代码
Python提供了众多的PDF支持库，本篇文章主要介绍了Python处理PDF及生成多层PDF实例代码，这样就能够实现图片扫描上来的内容也可以进行内容搜索的目标
2017-04-04
详解python __init__.py 和 __all__作用
导入文件夹包的时候，会运行写在该文件夹包下的__init__.py文件，这主要是__init__.py的作用，本文结合示例代码介绍了python __init__.py 和 __all__作用，感兴趣的朋友一起看看吧
2023-02-02
对python中的try、except、finally 执行顺序详解
今天小编就为大家分享一篇对python中的try、except、finally 执行顺序详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-02-02