nki.language.arange — AWS Neuron Documentation (original) (raw)

This document is relevant for: Inf2, Trn1, Trn2

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, Trn2