triton.experimental.gluon.language.arange

triton.experimental.gluon.language.arange(start, end, layout=None, _semantic=None)

Generate a sequence tensor with values in [start, end) using a specified layout.

Parameters:
  • start (int) – Inclusive start of the sequence.

  • end (int) – Exclusive end of the sequence.

  • layout (DistributedLayout) – The layout of the output tensor. Defaults to AutoLayout.

Returns:

A 1D tensor containing sequential values.

Return type:

tensor