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

ndarray

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.])