Mojo function
normalize_neg_index
normalize_neg_index(idx: Int, dim_size: Int) -> Int
Indices passed to gather and scatter ops may be negative. This performs a normalization so that they can be used to index into a buffer.
Returns val + dim if val < 0 else val
normalize_neg_index[type: DType, width: Int, out_type: DType = index](idx: SIMD[type, width], dim_size: Int) -> SIMD[out_type, width]
Indices passed to gather and scatter ops may be negative. This performs a normalization so that they can be used to index into a buffer.
Returns val + dim if val < 0 else val
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!