Python torch.flatten()函数案例详解

作者:Wanderer001 时间:2022-11-11 16:04:27 

先看函数参数:


torch.flatten(input, start_dim=0, end_dim=-1)

input: 一个 tensor,即要被“推平”的 tensor。

start_dim: “推平”的起始维度。

end_dim: “推平”的结束维度。

首先如果按照 start_dim 和 end_dim 的默认值,那么这个函数会把 input 推平成一个 shape 为 [n][n] 的tensor,其中 nn 即 input 中元素个数。

如果我们要自己设定起始维度和结束维度呢?

我们要先来看一下 tensor 中的 shape 是怎么样的:


t = torch.tensor([[[1, 2, 2, 1],
                  [3, 4, 4, 3],
                  [1, 2, 3, 4]],
                 [[5, 6, 6, 5],
                  [7, 8, 8, 7],
                  [5, 6, 7, 8]]])
print(t, t.shape)

运行结果:

tensor([[[1, 2, 2, 1],
        [3, 4, 4, 3],
        [1, 2, 3, 4]],

[[5, 6, 6, 5],
        [7, 8, 8, 7],
        [5, 6, 7, 8]]])
torch.Size([2, 3, 4])

我们可以看到,最外层的方括号内含两个元素,因此 shape 的第一个值是 2;类似地,第二层方括号里面含三个元素,shape 的第二个值就是 3;最内层方括号里含四个元素,shape 的第二个值就是 4。

示例代码:


x = torch.flatten(t, start_dim=1)
print(x, x.shape)

y = torch.flatten(t, start_dim=0, end_dim=1)
print(y, y.shape)

运行结果:

tensor([[1, 2, 2, 1, 3, 4, 4, 3, 1, 2, 3, 4],
       [5, 6, 6, 5, 7, 8, 8, 7, 5, 6, 7, 8]])
torch.Size([2, 12])

tensor([[1, 2, 2, 1],
       [3, 4, 4, 3],
       [1, 2, 3, 4],
       [5, 6, 6, 5],
       [7, 8, 8, 7],
       [5, 6, 7, 8]])
torch.Size([6, 4])

可以看到,当 start_dim = 11 而 end_dim = −1−1 时,它把第 11 个维度到最后一个维度全部推平合并了。而当 start_dim = 00 而 end_dim = 11 时,它把第 00 个维度到第 11 个维度全部推平合并了。pytorch中的 torch.nn.Flatten 类和 torch.Tensor.flatten 方法其实都是基于上面的 torch.flatten 函数实现的。

来源:https://blog.csdn.net/weixin_36670529/article/details/105668950

标签:Python,torch.flatten()
0
投稿

猜你喜欢

  • MySQL存储IP地址的方法

    2024-01-23 06:01:02
  • Javascript showModalDialog两个窗体之间传值

    2024-04-28 09:38:56
  • Dreamweaver如何制作会移动的广告条

    2010-10-20 20:04:00
  • Python编程实现的简单Web服务器示例

    2023-01-14 02:00:40
  • MySQL中几种数据统计查询的基本使用教程

    2024-01-13 22:29:56
  • Git里多种撤销操作的最佳方法

    2023-10-30 00:51:17
  • 关于ASP中脚本执行顺序的讲解

    2008-11-04 12:02:00
  • 在Golang中使用C语言代码实例

    2024-05-25 15:15:46
  • OpenCV Python实现拼图小游戏

    2021-07-30 18:10:34
  • python2.7安装opencv-python很慢且总是失败问题

    2021-01-10 06:11:15
  • vue-cli与webpack处理静态资源的方法及webpack打包的坑

    2024-05-09 09:39:07
  • python语言的优势是什么

    2022-04-15 18:45:25
  • 避免重复写代码的小函数

    2008-09-21 13:41:00
  • Python 稀疏矩阵-sparse 存储和转换

    2022-05-29 03:13:22
  • js绘制购物车抛物线动画

    2024-04-10 16:18:09
  • 微信JSSDK调用微信扫一扫功能的方法

    2024-04-29 13:46:02
  • js 实现数值的千分位及保存小数方法(推荐)

    2023-08-17 01:13:11
  • Django-Scrapy生成后端json接口的方法示例

    2021-07-16 18:46:46
  • python 点云地面点滤波-progressive TIN densification(PTD)算法介绍

    2023-05-26 17:50:55
  • 简单了解Javscript中兄弟ifream的方法调用

    2024-04-22 22:29:30
  • asp之家 网络编程 m.aspxhome.com