python:flatten()参数详解

python:flatten()参数详解

这篇博客主要写flatten()作用,及其参数的含义

  • flatten()是对多维数据的降维函数。
  • flatten(),默认缺省参数为0,也就是说flatten()和flatte(0)效果一样。
  • python里的flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维.也就是说,只保留dim之前的维度,其他维度的数据全都挤在dim这一维。
  • 比如一个数据的维度是 ( S 0 , S 1 , S 2......... , S n ) (S0,S1,S2.........,Sn) (S0,S1,S2.........,Sn), flatten(m)后的数据为 ( S 0 , S 1 , S 2 , . . . , S m − 2 , S m − 1 , S m ∗ S m + 1 ∗ S m + 2 ∗ . . . ∗ S n ) (S0,S1,S2,...,Sm-2,Sm-1,Sm*Sm+1*Sm+2*...*Sn) (S0S1S2...Sm2Sm1SmSm+1Sm+2...Sn)
  • 下面我们举个例子来理解一下flatten

比如我们随机定义一个维度为(2,3,4)的数据a

import torch
a = torch.rand(2,3,4)

a输出结果为:
在这里插入图片描述
a此时的维度为(2,3,4)

flatten()和flatten(0)效果一样,a这个数据从0维展开,就是( 2 ∗ 3 ∗ 4 2*3*4 234),维度就是(24)

b = a.flatten()
c = a.flatten(0)

b,c的结果如图:
在这里插入图片描述
b,c相等,维度均为 ( 2 ∗ 3 ∗ 4 ) (2*3*4) (234)也就是(24)

a从1维展开flatten(1),就是 ( 2 , 3 ∗ 4 ) (2,3*4) (2,34),也就是(2,12).

d = a.flatten(1)

d的结果如图所示:
在这里插入图片描述
a从一维展开后的数据d的维度为 ( 2 , 3 ∗ 4 ) (2,3*4) (2,34)也就是(2,12)

a若是从2维展开flatten(2),那就是(2,3,4)和之前没有变化

e = a.flatten(2)

e的结果如图所示:
在这里插入图片描述
e从二维展开后的数据d的维度为 ( 2 , 3 , 4 ) (2,3,4) (2,3,4)

总结

  • python里的flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维.也就是说,只保留dim之前的维度,其他维度的数据全都挤在dim这一维。
  • 比如一个数据的维度是 ( S 0 , S 1 , S 2 , S 3......... , S n ) (S0,S1,S2,S3.........,Sn) (S0,S1,S2,S3.........,Sn), flatten(m)后的数据为 ( S 0 , S 1 , S 2 , S 3 , . . . , S m − 2 , S m − 1 , S m ∗ S m + 1 ∗ S m + 2 ∗ . . . ∗ S n ) (S0, S1,S2,S3,...,Sm-2,Sm-1,Sm*Sm+1*Sm+2*...*Sn) (S0,S1S2S3...Sm2Sm1SmSm+1Sm+2...Sn)
  • 216
    点赞
  • 595
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 28
    评论
评论 28
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

皮皮宽

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值