Pytorch增加尺寸

据我所知,对于非单例维度,没有特殊的方法可以做到这一点。所以你的解决方案在我看来是正确的。

或者,在这里有效的方法是:

#given a tensor points of some size
pshape = list(points.size())
pshape[-1] += 1
z = torch.ones(pshape)
z[..., 0:-1] = points

在我的机器上,你的代码对小张量稍快,而我的建议对大张量稍快。不过,这两种情况的差异微乎其微。