Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches

◆ NN_arange()

Tensor * NN_arange ( size_t  ndim,
const size_t *  shape,
DataType  dtype,
float  start,
float  step 
)

Returns a tensor filled with increasing values.

Parameters
ndimnumber of dimensions
shapeshape of tensor
dtypedata type
startstarting value
stepincrement step size
Returns
Tensor