This document is relevant for: Inf2
, Trn1
, Trn1n
nki.language.arange#
- nki.language.arange(*args)[source]#
Return contiguous values within a given interval, used for indexing a tensor to define a tile.
((Similar to numpy.arange))
- arange can be called as:
arange(stop)
: Values are generated within the half-open interval[0, stop)
(the interval including zero, excluding stop).arange(start, stop)
: Values are generated within the half-open interval[start, stop)
(the interval including start, excluding stop).
This document is relevant for: Inf2
, Trn1
, Trn1n