megengine.functional.arange¶

arange(start=0, stop=None, step=1, *, dtype='float32', device=None)[source]¶

Returns evenly spaced values within the half-open interval [start, stop) as a one-dimensional tensor.

Note

This function cannot guarantee that the interval does not include the stop value in those cases where step is not an integer and floating-point rounding errors affect the length of the output tensor.

Parameters

start (Number) – if stop is specified, the start of interval (inclusive); otherwise, the end of the interval (exclusive). If stop is not specified, the default starting value is 0.
stop (Number) – the end of the interval.
step (Number) – the distance between two adjacent elements ( out[i+1] - out[i] ). Must not be 0 ; may be negative, this results i an empty tensor if stop >= start .

Keyword Arguments

dtype (Tensor.dtype, optional) – output tensor data type.
device (Tensor.device, optional) – device on which to place the created tensor.

See also

linspace

Return type

Tensor

Returns

A one-dimensional tensor containing evenly spaced values.

The length of the output tensor must be ceil((stop-start)/step) if stop - start and step have the same sign, and length 0 otherwise.

Examples

>>> F.arange(5)
Tensor([0. 1. 2. 3. 4.], device=xpux:0)
>>> F.arange(1, 4)
Tensor([1. 2. 3.], device=xpux:0)

megengine.functional

megengine.functional.linspace