jax.numpy.ones#
- jax.numpy.ones(shape, dtype=None, *, device=None)[source]#
Return a new array of given shape and type, filled with ones.
LAX-backend implementation of
numpy.ones()
.Original docstring below.
- Parameters:
- Returns:
out – Array of ones with the given shape, dtype, and order.
- Return type:
ndarray