[ torch 参数更多 ]torch.meshgrid

torch.meshgrid

torch.meshgrid(*tensors, indexing=None)

paddle.meshgrid

paddle.meshgrid(*args, **kargs)

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射

PyTorch PaddlePaddle 备注
tensors args 输入的 Tensor 列表，仅参数名不一致。
indexing - tensor 的组合模式。Paddle 无此参数，需要转写。

转写示例

indexing：tensor 的组合模式

# PyTorch 写法 (indexing 为‘ij’时)
torch.meshgrid(x, y, indexing='ij')

# Paddle 写法
paddle.meshgrid(x, y)

# PyTorch 写法 (indexing 为‘xy’时)
torch.meshgrid(x, y, indexing='xy')

# Paddle 写法
list([i.T for i in paddle.meshgrid(x, y)])