文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

pytorch中的view()函数怎么使用

2023-06-29 13:08

关注

这篇文章主要介绍了pytorch中的view()函数怎么使用的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇pytorch中的view()函数怎么使用文章都会有所收获,下面我们一起来看看吧。

一、普通用法 (手动调整size)

view()相当于reshape、resize,重新调整Tensor的形状。

import torcha1 = torch.arange(0,16)print(a1)# tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
a2 = a1.view(8, 2)a3 = a1.view(2, 8)a4 = a1.view(4, 4)print(a2)#tensor([[ 0,  1],#        [ 2,  3],#        [ 4,  5],#        [ 6,  7],#        [ 8,  9],#        [10, 11],#        [12, 13],#        [14, 15]])print(a3)#tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],#        [ 8,  9, 10, 11, 12, 13, 14, 15]])print(a4)#tensor([[ 0,  1,  2,  3],#        [ 4,  5,  6,  7],#        [ 8,  9, 10, 11],#        [12, 13, 14, 15]])

二、特殊用法:参数-1 (自动调整size)

view中一个参数定为-1,代表自动调整这个维度上的元素个数,以保证元素的总数不变。

v1 = torch.arange(0,16)print(v1)# tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])v2 = v1.view(-1, 16)v2# tensor([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15]])v2 = v1.view(-1, 8)v2# tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],#         [ 8,  9, 10, 11, 12, 13, 14, 15]])v2 = v1.view(-1, 4)v2#tensor([[ 0,  1,  2,  3],#        [ 4,  5,  6,  7],#        [ 8,  9, 10, 11],#        [12, 13, 14, 15]])v2 = v1.view(-1, 2)v2#tensor([[ 0,  1],#        [ 2,  3],#        [ 4,  5],#        [ 6,  7],#        [ 8,  9],#        [10, 11],#        [12, 13],#        [14, 15]])
v3 = v1.view(4*4, -1)v3# tensor([[ 0],#         [ 1],#         [ 2],#         [ 3],#         [ 4],#         [ 5],#         [ 6],#         [ 7],#         [ 8],#         [ 9],#         [10],#         [11],#         [12],#         [13],#         [14],#         [15]])

关于“pytorch中的view()函数怎么使用”这篇文章的内容就介绍到这里,感谢各位的阅读!相信大家对“pytorch中的view()函数怎么使用”知识都有一定的了解,大家如果还想学习更多知识,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯