torch.select(input, dim, index)
Paddle 无此 API,需要组合实现。
# Pytorch 写法
y = torch.select(a, dim=dim, index=index)
# Paddle 写法
y = paddle.index_select(a, index=paddle.to_tensor([index]), axis=dim).squeeze(dim)
torch.select(input, dim, index)
Paddle 无此 API,需要组合实现。
# Pytorch 写法
y = torch.select(a, dim=dim, index=index)
# Paddle 写法
y = paddle.index_select(a, index=paddle.to_tensor([index]), axis=dim).squeeze(dim)