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
See also
empty_like()
Return an empty array with shape and type of input.
zeros_like()
Return an array of zeros with shape and type of input.
full_like()
Return a new array with shape of input filled with value.
ones()
Return a new array setting values to one.
Examples
>>> x = np.arange(6) >>> x = x.reshape((2, 3)) >>> x array([[0, 1, 2], [3, 4, 5]]) >>> np.ones_like(x) array([[1, 1, 1], [1, 1, 1]])
>>> y = np.arange(3, dtype=float) >>> y array([0., 1., 2.]) >>> np.ones_like(y) array([1., 1., 1.])