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

torch.row_stack ¶

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

         

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

转写示例¶

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

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

         