Return dense multi-dimensional “meshgrid”.
LAX-backend implementation of numpy.mgrid. This is a convenience wrapper for functionality provided by jax.numpy.meshgrid() with sparse=False.
See also
jnp.ogrid: open/sparse version of jnp.mgrid
Examples
Pass [start:stop:step] to generate values similar to jax.numpy.arange():
>>> jnp.mgrid[0:4:1] Array([0, 1, 2, 3], dtype=int32)
Passing an imaginary step generates values similar to jax.numpy.linspace():
>>> jnp.mgrid[0:1:4j] Array([0. , 0.33333334, 0.6666667 , 1. ], dtype=float32)
Multiple slices can be used to create broadcasted grids of indices:
>>> jnp.mgrid[:2, :3]
Array([[[0, 0, 0],
[1, 1, 1]],
[[0, 1, 2],
[0, 1, 2]]], dtype=int32)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4