jax.numpy.linspace#

jax.numpy.linspace(start,stop,num=50,endpoint=True,retstep=False,dtype=None,axis=0,*,device=None)[source]#

Return evenly-spaced numbers within an interval.

JAX implementation ofnumpy.linspace().

Parameters:

• start (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – scalar or array of starting values.

• stop (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – scalar or array of stop values.

• num (int) – number of values to generate. Default: 50.

• endpoint (bool) – if True (default) then include thestop value in the result.If False, then exclude thestop value.

• retstep (bool) – If True, then return a(result,step) tuple, wherestep is theinterval between adjacent values inresult.

• axis (int) – integer axis along which to generate the linspace. Defaults to zero.

• device (Device |Sharding |None) – optionalDevice orShardingto which the created array will be committed.

• dtype (str |type[Any]|dtype |SupportsDType |None)

Returns:

• values is an array of evenly-spaced values fromstart tostop

• step is the interval between adjacent values.

Return type:

An arrayvalues, or a tuple(values,step) ifretstep is True, where

See also

• jax.numpy.arange(): GenerateN evenly-spaced values given a startingpoint and a step

• jax.numpy.logspace(): Generate logarithmically-spaced values.

• jax.numpy.geomspace(): Generate geometrically-spaced values.

Examples

List of 5 values between 0 and 10:

>>>jnp.linspace(0,10,5)Array([ 0. ,  2.5,  5. ,  7.5, 10. ], dtype=float32)

List of 8 values between 0 and 10, excluding the endpoint:

>>>jnp.linspace(0,10,8,endpoint=False)Array([0.  , 1.25, 2.5 , 3.75, 5.  , 6.25, 7.5 , 8.75], dtype=float32)

List of values and the step size between them

>>>vals,step=jnp.linspace(0,10,9,retstep=True)>>>valsArray([ 0.  ,  1.25,  2.5 ,  3.75,  5.  ,  6.25,  7.5 ,  8.75, 10.  ],      dtype=float32)>>>stepArray(1.25, dtype=float32)

Multi-dimensional linspace:

>>>start=jnp.array([0,5])>>>stop=jnp.array([5,10])>>>jnp.linspace(start,stop,5)Array([[ 0.  ,  5.  ],       [ 1.25,  6.25],       [ 2.5 ,  7.5 ],       [ 3.75,  8.75],       [ 5.  , 10.  ]], dtype=float32)