jax.numpy.onesΒΆ
-
jax.numpy.
ones
(shape, dtype=None)[source]ΒΆ Return a new array of given shape and type, filled with ones.
LAX-backend implementation of
ones()
. Original docstring below.- Parameters
shape (int or sequence of ints) β Shape of the new array, e.g.,
(2, 3)
or2
.dtype (data-type, optional) β The desired data-type for the array, e.g., numpy.int8. Default is numpy.float64.
- Returns
out β Array of ones with the given shape, dtype, and order.
- Return type
See also
ones_like()
Return an array of ones with shape and type of input.
empty()
Return a new uninitialized array.
zeros()
Return a new array setting values to zero.
full()
Return a new array of given shape filled with value.
Examples
>>> np.ones(5) array([1., 1., 1., 1., 1.])
>>> np.ones((5,), dtype=int) array([1, 1, 1, 1, 1])
>>> np.ones((2, 1)) array([[1.], [1.]])
>>> s = (2,2) >>> np.ones(s) array([[1., 1.], [1., 1.]])