meshgrid

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

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

参数：

  • * args （Tensor|Tensor数组）- 输入变量为 k 个一维张量，形状分别为(N1,), (N2,), …, (Nk, )。支持数据类型为float32，float64，int32，int64。

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

返回： k 个 k 维张量，每个张量的形状均为(N1, N2, …, Nk)。

返回类型： Tensor，k个Tensor，每个Tensor的形状分别为N1, N2, … , Nk。

代码示例

  1. import paddle
  3. x = paddle.randint(low=0, high=100, shape=[100])
  4. y = paddle.randint(low=0, high=100, shape=[200])
  6. grid_x, grid_y = paddle.meshgrid(x, y)
  8. print(grid_x.shape)
  9. print(grid_y.shape)
  11. #the shape of res_1 is (100, 200)
  12. #the shape of res_2 is (100, 200)