jax.numpy.ones_like¶
-
jax.numpy.ones_like(a, dtype=None, shape=None)[source]¶ Return an array of ones with the same shape and type as a given array.
LAX-backend implementation of
ones_like().Original docstring below.
- Parameters
a (array_like) – The shape and data-type of a define these same attributes of the returned array.
dtype (data-type, optional) – Overrides the data type of the result.
shape (int or sequence of ints, optional.) – Overrides the shape of the result. If order=’K’ and the number of dimensions is unchanged, will try to keep order, otherwise, order=’C’ is implied.
- Returns
out – Array of ones with the same shape and type as a.
- Return type