jax.numpy.empty_likeΒΆ
-
jax.numpy.
empty_like
(a, dtype=None, shape=None)ΒΆ Return an array of zeros with the same shape and type as a given array.
LAX-backend implementation of
zeros_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 zeros with the same shape and type as a.
- Return type
See also
empty_like()
Return an empty array with shape and type of input.
ones_like()
Return an array of ones with shape and type of input.
full_like()
Return a new array with shape of input filled with value.
zeros()
Return a new array setting values to zero.
Examples
>>> x = np.arange(6) >>> x = x.reshape((2, 3)) >>> x array([[0, 1, 2], [3, 4, 5]]) >>> np.zeros_like(x) array([[0, 0, 0], [0, 0, 0]])
>>> y = np.arange(3, dtype=float) >>> y array([0., 1., 2.]) >>> np.zeros_like(y) array([0., 0., 0.])