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

更新时间：2019年11月05日 11:48:25 作者：MY Qi

这篇文章主要介绍了pytorch torch.expand和torch.repeat的区别详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

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计算斗牛游戏概率算法,简单介绍了斗牛游戏的原理并结合具体实例形式分析了相关的游戏概率算法,需要的朋友可以参考下
2017-09-09
用python删除文件夹中的重复图片(图片去重)
近期在整理相册的时候，发现相册中有许多重复图片，人工一张张筛查删除太枯燥，便写下这个python脚本，用于删除文件夹下重复的图片
2021-05-05
Python获取对象属性的几种方式小结
这篇文章主要介绍了Python获取对象属性的几种方式小结，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-03-03
Python项目打包成二进制的方法
这篇文章主要介绍了Python项目打包成二进制的方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2020-12-12
如何在Python项目中引入日志
在开发一些大型项目的时候，都会使用日志来记录项目运行时产生的信息，以备出错时定位分析和从日志信息中提取数据统计分析等。在 Python 中使用 logging 内置模块即可对项目进行日志的配置。
2021-05-05
python操作手机app的实现步骤
本文主要介绍了python操作手机app的实现步骤，本文将结合实例代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-07-07
Python利用Pillow(PIL)库实现验证码图片的全过程
这篇文章主要给大家介绍了关于Python利用Pillow(PIL)库实现验证码图片的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-10-10
python实战之90行代码写个猜数字游戏
这篇文章主要介绍了python实战之90行代码写个猜数字,文中有非常详细的代码示例,对正在学习python的小伙伴们有很大的帮助,需要的朋友可以参考下
2021-04-04
python深度学习标准库使用argparse调参
这篇文章主要为大家介绍了python深度学习标准库使用argparse调参实例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-06-06
Pytorch实现tensor序列化和并行化的示例详解
这篇文章主要介绍了Pytorch实现tensor序列化和并行化,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,感兴趣的同学们下面随着小编来一起学习学习吧
2023-12-12