jax.numpy.ogrid

jax.numpy.ogrid = <jax._src.numpy.index_tricks._Ogrid object>

返回開放式多維「meshgrid」。

numpy.ogrid 的 LAX 後端實作。這是 jax.numpy.meshgrid() 與 sparse=True 提供功能的便利包裝器。

另請參閱

jnp.mgrid：jnp.ogrid 的密集版本

範例

傳遞 [start:stop:step] 以生成類似於 jax.numpy.arange() 的值

>>> jnp.ogrid[0:4:1]
Array([0, 1, 2, 3], dtype=int32)

傳遞虛數步長會生成類似於 jax.numpy.linspace() 的值

>>> jnp.ogrid[0:1:4j]
Array([0.        , 0.33333334, 0.6666667 , 1.        ], dtype=float32)

多個切片可用於建立稀疏索引網格

>>> jnp.ogrid[:2, :3]
[Array([[0],
        [1]], dtype=int32),
 Array([[0, 1, 2]], dtype=int32)]