pytorch 计算ConvTranspose1d输出特征大小方式
作者:明月几时有. 时间:2021-08-14 10:16:16
问题:如何经过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()
来源:https://blog.csdn.net/weixin_35576881/article/details/90075133
标签:pytorch,ConvTranspose1d,特征
0
投稿
猜你喜欢
Django日志及中间件模块应用案例
2021-06-02 22:16:50
php文件怎么打开 如何执行php文件
2023-11-15 05:51:46
vue 弹框产生的滚动穿透问题的解决
2024-04-26 17:40:54
PHP投票系统防刷票判断流程分析
2023-09-27 13:34:03
ADO.NET 连接数据库字符串小结(Oracle、SqlServer、Access、ODBC)
2024-01-22 07:43:47
Python中JSON的使用方法(超详细)
2023-11-15 23:04:19
plt.figure()参数使用详解及运行演示
2021-11-01 04:52:19
python k-近邻算法实例分享
2022-03-26 14:47:17
vue中使用iframe嵌入网页,页面可自适应问题
2024-04-26 17:38:24
Pyhhon之Pygame的Font文本和字体
2021-08-14 13:00:33
博客使用标签是迟早的事
2008-10-31 12:55:00
python处理DICOM并计算三维模型体积
2022-07-16 09:24:29
Python用模块pytz来转换时区
2021-05-27 20:09:32
用于业余项目的8个优秀Python库
2022-04-04 09:44:51
详解MySQL高可用MMM搭建方案及架构原理
2024-01-18 07:28:09
python3读取autocad图形文件.py实例
2022-03-11 09:08:45
基于python 处理中文路径的终极解决方法
2023-07-21 00:47:09
sql2005 日志清理 SQL2005压缩清除日志的方法
2024-01-28 12:59:25
聊一聊MyISAM和InnoDB的区别
2024-01-21 00:55:33
使用jQuery.form.js/springmvc框架实现文件上传功能
2024-04-09 19:46:12