jax.numpy.full

jax.numpy.full(shape, fill_value, dtype=None)

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

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

Parameters

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

• fill_value (scalar or array_like) – Fill value.

• dtype (data-type, optional) –

  The desired data-type for the array The default, None, means

  np.array(fill_value).dtype.

Returns

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

Return type

ndarray

See also

full_like()

Return a new array with shape of input filled with value.

empty()

Return a new uninitialized array.

ones()

Return a new array setting values to one.

zeros()

Return a new array setting values to zero.

Examples

>>> np.full((2, 2), np.inf)
array([[inf, inf],
       [inf, inf]])
>>> np.full((2, 2), 10)
array([[10, 10],
       [10, 10]])

>>> np.full((2, 2), [1, 2])
array([[1, 2],
       [1, 2]])