Python：flatten()函数用法

介绍：

flatten()是对多维数据的降维函数
flatten(),默认缺省参数为0
适用：numpy对象，即数组array或者矩阵MAT，普通的list列表不可以
出于：flatten是numpy.ndarray.flatten的一个函数
详细：python里的flatten(dim)表示，从第dim个维度开始展开，将后面的维度转化为一维.也就是说，只保留dim之前的维度，其他维度的数据全都挤在dim这一维。

举例：

import torch
a = torch.rand(2,3,4) # 0，1，2维
print(a.size())# torch.Size([2, 3, 4])

flatten()函数的默认缺省参数为0，即可以理解为将所有维度数相乘，恢复原始数据

b = a.flatten(0)  # 2*3*4
print(b.size()) # torch.Size([24])

保留第0维的维度，其他维度上的数字转成一维

b = a.flatten(1) # 2，3*4
print(b.size())# torch.Size([2, 12])

保留第0，1维的维度，其他维度上的数字转成一维

b = a.flatten(2) # 2，3，4(除了第0，1维，其他维只有4)
print(b.size())# torch.Size([2, 3, 4])

为深入理解，定义一个5维的数据。保留第0，1，2维的维度，将其他维度上的数字转成一维。可以直接想到，5维的数据，即包含0，1，2，3，4维，保留三个维度的数，则5维数据变成了4维。若保留0，1维，则返回一个3维的数据。

import torch
a = torch.rand(2,3,4,5,6)  # 2，3，4，5*6
b = a.flatten(3)
b.size()# torch.Size([2, 3, 4, 30])

参考：
皮皮宽：python：flatten()参数详解

Mingsheng Zhang：flatten()函数用法

Python：flatten()函数用法相关推荐

python之函数用法islower()
# -*- coding: utf-8 -*- #python 27 #xiaodeng #python之函数用法islower() #http://www.runoob.com/python/att ...
python之函数用法startswith()
# -*- coding: utf-8 -*- #python 27 #xiaodeng #python之函数用法startswith() #http://www.runoob.com/python/ ...
python之函数用法__getitem__()
# -*- coding: utf-8 -*- #python 27 #xiaodeng #python之函数用法__getitem__() #http://www.cnblogs.com/hongf ...
python之函数用法capitalize()
# -*- coding: utf-8 -*- #python 27 #xiaodeng #python之函数用法capitalize()#capitalize() #说明:将字符串的第一个字母变成大 ...
python之函数用法isupper()
# -*- coding: utf-8 -*- #python 27 #xiaodeng #python之函数用法isupper() #http://www.runoob.com/python/att ...
flatten()函数用法
flatten()函数用法 flatten是numpy.ndarray.flatten的一个函数,即返回一个折叠成一维的数组.但是该函数只能适用于numpy对象,即array或者mat,普通的list ...
Python range() 函数用法细解（附猴子吃桃问题引例）
Python range() 函数用法细解 --步入大学,好多计算机系的同学可能专业课学c或者java,但是随着python变成语言的崛起,往往一些学校也会要求学生们学习Python这门编程语言(我们 ...
python print函数用法_Python3.2中Print函数用法实例详解
本文实例讲述了Python3.2中Print函数用法.分享给大家供大家参考.具体分析如下: 1. 输出字符串 >>> strHello = 'Hello World' >> ...
python所有函数用法_python函数用法总结
空函数如果想定义一个什么事也不做的空函数,可以用pass语句: def nop(): pass pass语句什么都不做,那有什么用?实际上pass可以用来作为占位符,比如现在还没想好怎么写函数的代码 ...

Python：flatten()函数用法

介绍：

举例：

flatten()函数的默认缺省参数为0，即可以理解为将所有维度数相乘，恢复原始数据

保留第0维的维度，其他维度上的数字转成一维

保留第0，1维的维度，其他维度上的数字转成一维

为深入理解，定义一个5维的数据。保留第0，1，2维的维度，将其他维度上的数字转成一维。可以直接想到，5维的数据，即包含0，1，2，3，4维，保留三个维度的数，则5维数据变成了4维。若保留0，1维，则返回一个3维的数据。

Python：flatten()函数用法相关推荐

最新文章

热门文章