jax.numpy.ones

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

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) 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 ones with the given shape, dtype, and order.

Return type

ndarray

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