torch.meshgrid

torch.meshgrid

torch.meshgrid(*tensors, indexing=None)

paddle.meshgrid

paddle.meshgrid(*args, **kargs)

其中 Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数差异

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | tensors | args | 输入的 Tensor 列表 | | indexing | - | tensor 的组合模式 |

转写示例

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

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

# 当 indexing 不是‘ij’时,暂时无法转写