已安装 Torch，但未找到 repeat_interleave

Question

我已经安装了 PyTorch。 import torch 运行没有错误。但是找不到函数torch.repeat_interleave()：

x = torch.tensor([1, 2, 3])
x.repeat_interleave(2)

给AttributeError: 'Tensor' object has no attribute 'repeat_interleave'

为什么？

Answer 1

torch.repeat_interleave 运算符是在pytorch 的1.1.0 版本中引入的，所以请考虑更新到1.1.0+ 版本的pytorch 以顺利使用此方法。