jax.numpy.fullΒΆ
-
jax.numpy.
full
(shape, fill_value, dtype=None)[source]ΒΆ 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)
or2
.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
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]])