[ 组合替代实现 ]torch.Tensor.select

torch.Tensor.select

torch.Tensor.select(dim, index)

Paddle 无此 API,需要组合实现。

转写示例

# PyTorch 写法
y = a.select(dim=dim, index=index)

# Paddle 写法
y = paddle.index_select(a, index=paddle.to_tensor([index]), axis=dim).squeeze(dim)