mindspore.numpy.mgrid

查看源文件
mindspore.numpy.mgrid()

返回一个密集矩阵 NdGrid 实例,其中 sparse=False 。 输出数组的维度和数量等于索引维度的数量。如果 step 不是一个复数,则 stop 不包括在内。然而,如果 step 是复数(例如5j),那么它的整数部分被解释为指定要在 startstop 之间创建的点的数量,其中 stop 包括在内。

说明

在graph模式下不受支持。与Numpy不同,如果 step 是一个带有实数分量的复数,则 step 被处理为等效于 int(abs(step))

返回:

元素为Tensor或Tensor的Tuple,返回 meshgrid 网格点坐标数组。

异常:
  • TypeError - 如果切片索引不是整数。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> output = np.mgrid[0:5, 0:5]
>>> print(output)
[[[0 0 0 0 0]
[1 1 1 1 1]
[2 2 2 2 2]
[3 3 3 3 3]
[4 4 4 4 4]]
[[0 1 2 3 4]
[0 1 2 3 4]
[0 1 2 3 4]
[0 1 2 3 4]
[0 1 2 3 4]]]
>>> output = mgrid[-1:1:5j]
>>> print(output)
[-1.  -0.5  0.   0.5  1. ]