jax.numpy.ones#
- jax.numpy.ones(shape, dtype=None)[source]#
Return a new array of given shape and type, filled with ones.
LAX-backend implementation of
numpy.ones()
.Original docstring below.
- Parameters
shape (int or sequence of ints) – Shape of the new array, e.g.,
(2, 3)
or2
.dtype (data-type, optional) – The desired data-type for the array, e.g., numpy.int8. Default is numpy.float64.
- Returns
out – Array of ones with the given shape, dtype, and order.
- Return type
ndarray