meshgrid

paddle. meshgrid ( *args, **kargs ) [源代码]

对每个 Tensor 做扩充操作。输入是 Tensor 或者包含 Tensor 的列表,包含 k 个一维 Tensor,输出 k 个 k 维 Tensor。

参数

  • args (Tensor|Tensor 数组) - 输入变量为 k 个一维 Tensor,形状分别为(N1,), (N2,), ..., (Nk, )。支持数据类型为 bfloat16、float16、float32、float64、int32、int64、complex64 和 complex128。

  • kargs (可选) - 目前只接受 name 参数(str),具体用法请参见 Name,一般无需设置,默认值为 None。

返回

k 个 k 维 Tensor,每个形状均为(N1, N2, ..., Nk)。

代码示例

>>> import paddle

>>> x = paddle.randint(low=0, high=100, shape=[100])
>>> y = paddle.randint(low=0, high=100, shape=[200])

>>> grid_x, grid_y = paddle.meshgrid(x, y)

>>> print(grid_x.shape)
[100, 200]
>>> print(grid_y.shape)
[100, 200]

使用本API的教程文档