triton.language.arange

triton.language.arange(start, end)

返回半开区间[start, end)内的连续值。end - start必须小于或等于 TRITON_MAX_TENSOR_NUMEL = 1048576

Parameters:

• start (int32) - 区间的起始值。必须是2的幂。

• end (int32) – 区间的结束值。必须是大于start的2的幂。