欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

numpy的repeat和pytorch的repeat

时间:2023-08-16
numpy的repeat


重复数组中的元素

样例1

从某一个维度复制,如下面从第一维度复制,(2,3)的张量复制后就是(4,3)

x = np.array([1,2,3,4,5,6]).reshape(2,3)print(x)print("===repeat====")# 也可以写作为# x = np.repeat(x, 2, axis=0)x = x.repeat(2, axis=0)print(x, x.shape)

如果复制第二个维度呢,那么(2,3)的张量复制后就是(2,6),但是复制内容是一个一个复制,比如,1,2,3的复制是1,1,2,2,3,3而不是1,2,3,1,2,3

x = np.array([1,2,3,4,5,6]).reshape(2,3)print(x)print("===repeat====")# 也可以写作为# x = np.repeat(x, 2, axis=1)x = x.repeat(2, axis=1)print(x, x.shape)

样例2

如果不指定axis会怎么样呢? 是把整个Tensor压成一维,然后repeat

x = np.array([1,2,3,4,5,6]).reshape(2,3)print(x)print("===repeat====")# 也可以写作为# x = np.repeat(x, 2)x = x.repeat(2)print(x, x.shape)

样例3

使用repeat初始化数组
第一个参数是重复的数,后一个参数是重复的个数

print("===repeat====")x=np.repeat(2,3)print(x)

样例4

在一个维度上重复的次数不一致

x = np.array([1,2,3,4,5,6]).reshape(2,3)print(x)print("===repeat====")# 也可以写作为# x = np.repeat(x, [1,2], axis=0)x = x.repeat([1,2], axis=0)print(x, x.shape)


numpy的tile

tile跟repeat功能基本一样,主要的区别是tile的复制是单个维度数据的整体复制;
但是它没有axis参数可以指定维度

样例1

x = np.array([1,2,3,4,5,6]).reshape(2,3)print(x)print("===repeat====")# 注意不可以写作x.tilex = np.tile(x, 2)print(x, x.shape)


主要区别就是复制方式不同,而且只能作用于最后一维

=======================================================

torch的repeat 样例1

这个样例的操作类似numpy.tile(x, 2)

x = np.array([1,2,3,4,5,6]).reshape(2,3)x1 = torch.from_numpy(x)print(x1)print("===repeat====")x1 = x1.repeat(1,2)print(x1, x1.shape)

样例2

这个样例有点类似numpy.repeat(x, 2, axis=0),但是复制的细节略有不同,请注意。

x = np.array([1,2,3,4,5,6]).reshape(2,3)x1 = torch.from_numpy(x)print(x1)print("===repeat====")x1 = x1.repeat(2,1)print(x1, x1.shape)

样例3

如果repeat的维度多于原始张量的维度,那么会自动做阔维

x = np.array([1,2,3,4,5,6]).reshape(2,3)x1 = torch.from_numpy(x)print(x1)print("===repeat====")x1 = x1.repeat(2,2,2)print(x1, x1.shape)

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。