pytorch 中repeat函数

沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。
扩展(expand)张量不会分配新的内存,只是在存在的张量上创建一个新的视图(view),一个大小(size)等于1的维度扩展到更大的尺寸。repeat沿着特定的维度重复这个张量,和expand()不同的是,这个函数拷贝张量的数据。

import torch a=torch.linspace(0,12,13)

print(a) print(a.shape)

pytorch 中repeat函数
文章图片

b=a.repeat(13,1) print(b) print(b.shape)

【pytorch 中repeat函数】pytorch 中repeat函数
文章图片

c=b.repeat(3,1,1) print(c) print(c.shape)

pytorch 中repeat函数
文章图片

    推荐阅读