cupy.meshgrid#

cupy.meshgrid(*xi, **kwargs)[源代码]#

从坐标向量返回坐标矩阵。

给定一维坐标数组 x1, x2, ... , xn,此函数创建 N 维网格。

对于长度为 Ni = len(xi) 的一维数组 x1, x2, ... , xn,如果索引方式为 ‘ij’,则此函数返回形状为 (N1, N2, N3, ..., Nn) 的数组;如果索引方式为 ‘xy’,则返回形状为 (N2, N1, N3, ..., Nn) 的数组。

与 NumPy 不同,CuPy 目前仅支持一维数组作为输入。

参数:
  • xi (tuple of ndarrays) – 表示网格坐标的一维数组。

  • indexing ({'xy', 'ij'}, optional) – 输出数组的笛卡尔坐标系索引 (‘xy’,默认) 或矩阵索引 (‘ij’)。

  • sparse (bool, optional) – 如果为 True,则返回稀疏网格以节省内存。默认值为 False

  • copy (bool, optional) – 如果为 False,则返回原始数组的视图。默认值为 True

返回值:

cupy.ndarray 列表

另请参阅

numpy.meshgrid()