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