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

Question

我正在寻找一个类似于 tf 的 tile 函数的 pytorch 函数。我看到PyTorch 曾经有一个 tile 功能，但显然它被删除了。

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

标签： tensorflowpytorchtensor

Answer 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])