CNN记录】pytorch中flatten函数

pytorch原型

python 复制代码
torch.flatten(input, start_dim=0, end_dim=- 1)

作用:将连续的维度范围展平维张量,一般写再某个nn后用于对输出处理,

参数:

start_dim:开始的维度

end_dim:终止的维度,-1为最后一个轴

默认值时展平为1维

例子

1、默认参数

python 复制代码
input = torch.randn(2, 3, 4, 5)
output = torch.flatten(input)
输出维:torch.Size([120])

2、设置参数

python 复制代码
input = torch.randn(2, 3, 4, 5)

output = torch.flatten(input,1)
输出shape为:torch.Size([2, 60])

output = torch.flatten(input,1,2)
输出shape为:torch.Size([2, 12, 5])
相关推荐
LiJieNiub14 小时前
基于 PyTorch 实现 MNIST 手写数字识别
pytorch·深度学习·学习
chxin1401614 小时前
Transformer注意力机制——动手学深度学习10
pytorch·rnn·深度学习·transformer
MYX_30914 小时前
第五章 神经网络的优化
pytorch·深度学习·神经网络·学习
麦麦大数据17 小时前
F024 CNN+vue+flask电影推荐系统vue+python+mysql+CNN实现
vue.js·python·cnn·flask·推荐算法
算法打盹中17 小时前
计算机视觉:卷积神经网络(CNN)图像分类从像素与色彩通道基础到特征提取、池化及预测
图像处理·神经网络·计算机视觉·cnn·图像分类
久未17 小时前
Pytorch autoload机制自动加载树外扩展(Autoload Device Extension)
人工智能·pytorch·python
西柚小萌新19 小时前
【深入浅出PyTorch】--7.1.PyTorch可视化1
人工智能·pytorch·python
一车小面包20 小时前
Transformers中从 logits 本质到问答系统中的字符定位机制
pytorch·python·深度学习
西柚小萌新1 天前
【深入浅出PyTorch】--上采样+下采样
人工智能·pytorch·python
dlraba8021 天前
PyTorch 模型部署实战:用 Flask 搭图像分类 API
pytorch·分类·flask