pytorch torch.expand和torch.repeat的区别详解


Posted in Python onNovember 05, 2019

1.torch.expand

函数返回张量在某一个维度扩展之后的张量,就是将张量广播到新形状。函数对返回的张量不会分配新内存,即在原始张量上返回只读视图,返回的张量内存是不连续的。类似于numpy中的broadcast_to函数的作用。如果希望张量内存连续,可以调用contiguous函数。

例子:

import torch

x = torch.tensor([1, 2, 3, 4])
xnew = x.expand(2, 4)
print(xnew)

输出:

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

2.torch.repeat

torch.repeat用法类似np.tile,就是将原矩阵横向、纵向地复制。与torch.expand不同的是torch.repeat返回的张量在内存中是连续的。

例子1:

将张量横向的复制

import torch

x = torch.tensor([1, 2, 3])
xnew = x.repeat(1,3)
print(xnew)

输出:

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

例子2:

将张量纵向的复制

import torch

x = torch.tensor([1, 2, 3])
xnew = x.repeat(3,1)
print(xnew)

输出:

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python网络编程之读取网站根目录实例
Sep 30 Python
python多重继承实例
Oct 11 Python
10个易被忽视但应掌握的Python基本用法
Apr 01 Python
Python实现把xml或xsl转换为html格式
Apr 08 Python
python创建列表并给列表赋初始值的方法
Jul 28 Python
[原创]教女朋友学Python(一)运行环境搭建
Nov 29 Python
Python XML转Json之XML2Dict的使用方法
Jan 15 Python
pytorch torch.expand和torch.repeat的区别详解
Nov 05 Python
Pytorch在dataloader类中设置shuffle的随机数种子方式
Jan 14 Python
PyCharm取消波浪线、下划线和中划线的实现
Mar 03 Python
python等待10秒执行下一命令的方法
Jul 19 Python
Python实现异步IO的示例
Nov 05 Python
Python socket模块ftp传输文件过程解析
Nov 05 #Python
python3.6、opencv安装环境搭建过程(图文教程)
Nov 05 #Python
Python socket模块方法实现详解
Nov 05 #Python
基于python3 的百度图片下载器的实现代码
Nov 05 #Python
Python hashlib模块加密过程解析
Nov 05 #Python
python set集合使用方法解析
Nov 05 #Python
基于python2.7实现图形密码生成器的实例代码
Nov 05 #Python
You might like
一些常用的php函数
2006/12/06 PHP
程序员编程十条戒律
2009/07/09 PHP
php设计模式 Proxy (代理模式)
2011/06/26 PHP
基于PHP 面向对象之成员方法详解
2013/05/04 PHP
php中array_column函数简单实现方法
2016/07/11 PHP
php获取远程图片并下载保存到本地的方法分析
2016/10/08 PHP
js 异步处理进度条
2010/04/01 Javascript
js之onload事件的一点使用心得
2013/08/14 Javascript
浅谈javascript六种数据类型以及特殊注意点
2013/12/20 Javascript
jquery带有索引按钮且自动轮播切换特效代码分享
2015/09/15 Javascript
javascript每日必学之循环
2016/02/19 Javascript
判断JS对象是否拥有某属性的方法推荐
2016/05/12 Javascript
Jquery元素追加和删除的实现方法
2016/05/24 Javascript
原生JS实现在线问卷调查投票特效
2017/01/03 Javascript
jQuery内容筛选选择器实例代码
2017/02/06 Javascript
NodeJs下的测试框架Mocha的简单介绍
2017/02/22 NodeJs
angularJS模态框$modal实例代码
2017/05/27 Javascript
js实现购物车功能
2018/06/12 Javascript
基于Node的Axure文件在线预览的实现代码
2019/08/28 Javascript
element-ui点击查看大图的方法示例
2020/12/14 Javascript
JavaScript中展开运算符及应用的实例代码
2021/01/14 Javascript
[02:41]DOTA2英雄基础教程 谜团
2013/12/10 DOTA
Python简单连接MongoDB数据库的方法
2016/03/15 Python
Python MD5加密实例详解
2017/08/02 Python
EM算法的python实现的方法步骤
2018/01/02 Python
简单实现python数独游戏
2018/03/30 Python
Jupyter Notebook折叠输出的内容实例
2020/04/22 Python
为什么称python为胶水语言
2020/06/16 Python
20年同学聚会邀请函
2014/02/04 职场文书
大学生就业自我推荐信
2014/05/10 职场文书
酒店工程部岗位职责
2015/02/12 职场文书
人口与计划生育责任书
2015/05/09 职场文书
2016清明节森林防火广播稿
2015/12/17 职场文书
美甲店的创业计划书模板
2019/08/23 职场文书
Python制作春联的示例代码
2022/01/22 Python
centos7安装mysql5.7经验记录
2022/05/02 Servers