BendMlTensorArray.ASet.J: def of bend-ml-tensor-array@0.1.19.0
Import: import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
Lines 2034-2036 of bend-ml-tensor-array@0.1.19.0/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-tensor-array@0.1.19.0/main.bend&start_line=2034&end_line=2036

import Base

# the masked index and the first branch, as Array.swap.at and Array.get.at compute them from a size n
def ASet.J(+i: U32, +n: U32) -> U32:
  U32.and(i, U32.sub(n, 1))

Uses (3):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.and
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.sub