[ 组合替代实现 ]torch.vstack¶

torch.vstack ¶

torch.vstack(tensors, *, out=None)

按垂直方向拼接张量; Paddle 无此 API，需要组合实现。

# Pytorch 写法
y = torch.vstack((a, b))

# Paddle 写法
if a.ndim == 1:
    y = paddle.stack((a, b))
else:
    y = paddle.concat((a, b))