mars.tensor.mgrid

mars.tensor.mgrid = <mars.tensor.lib.index_tricks.nd_grid object>

构建一个多维“网格”。

grid = nd_grid() 创建一个实例，当被索引时将返回一个网格。输出数组的维度和数量与索引维度的数量相等。如果步长不是复数，则终止不包含在内。

然而，如果步长是一个 复数（例如 5j），那么它的模的整数部分被解释为指定在开始和停止值之间创建的点数，其中停止值 是包含的。

如果使用参数 sparse=True 实例化，则网格是开放的（或未展开），以便返回的每个参数只有一个维度大于 1。

Parameters

稀疏 (布尔值, 可选) – 网格是否是稀疏的。默认为 False。

备注

在Mars.tensor命名空间中提供了两个nd_grid的实例，mgrid和ogrid：

mgrid = nd_grid(sparse=False)
ogrid = nd_grid(sparse=True)

用户应使用这些预定义的实例，而不是直接使用 nd_grid。

示例

>>> import mars.tensor as mt

>>> mgrid = mt.lib.index_tricks.nd_grid()
>>> mgrid[0:5,0:5]
array([[[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]]])
>>> mgrid[-1:1:5j]
array([-1. , -0.5,  0. ,  0.5,  1. ])

>>> ogrid = mt.lib.index_tricks.nd_grid(sparse=True)
>>> ogrid[0:5,0:5]
[array([[0],
        [1],
        [2],
        [3],
        [4]]), array([[0, 1, 2, 3, 4]])]