jax.numpy.empty

jax.numpy.empty(shape, dtype=None)

Return a new array of given shape and type, filled with zeros.

LAX-backend implementation of zeros(). Original docstring below.

zeros(shape, dtype=float, order=’C’)

Parameters

• shape (int or tuple of ints) – Shape of the new array, e.g., (2, 3) or 2.

• dtype (data-type, optional) – The desired data-type for the array, e.g., numpy.int8. Default is numpy.float64.

Returns

out – Array of zeros with the given shape, dtype, and order.

Return type

ndarray

See also

zeros_like()

Return an array of zeros with shape and type of input.

empty()

Return a new uninitialized array.

ones()

Return a new array setting values to one.

full()

Return a new array of given shape filled with value.

Examples

>>> np.zeros(5)
array([ 0.,  0.,  0.,  0.,  0.])

>>> np.zeros((5,), dtype=int)
array([0, 0, 0, 0, 0])

>>> np.zeros((2, 1))
array([[ 0.],
       [ 0.]])

>>> s = (2,2)
>>> np.zeros(s)
array([[ 0.,  0.],
       [ 0.,  0.]])

>>> np.zeros((2,), dtype=[('x', 'i4'), ('y', 'i4')]) # custom dtype
array([(0, 0), (0, 0)],
      dtype=[('x', '<i4'), ('y', '<i4')])