寻找一个pytorch函数来重复一个向量

【问题标题】：Looking for a pytorch function to repeat a vector寻找一个pytorch函数来重复一个向量
【发布时间】：2021-05-21 19:02:57
【问题描述】：

我正在寻找一个类似于 tf 的 tile 功能的 pytorch 功能。我看到了PyTorch used to have a tile function，但显然它已被删除。

我正在寻找的功能示例：假设我有一个尺寸为 (1,1,1,1000) 的张量，我想重复几次，所以我得到一个 (1,40,40,1000) 的张量。

【问题讨论】：

【解决方案1】：

Torch 张量有一个 repeat() 方法，因此：

a = torch.rand((1, 1, 1, 1000))
b = a.repeat(1, 40, 40, 1)
b.shape  # Gives torch.Size([1, 40, 40, 1000])

【讨论】：