def right_start

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:269

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix

269def right_start(+layout: Layout, +position: Nat) -> Nat:
270  Layout{lo, +ro, +inner, +rows, +columns, l1, l2, +r1, +r2, +o2} = layout
271  +batch = Nat.div(position, Nat.mul(rows, columns))
272  Nat.add(Nat.mul(batch_number(r1, r2, Nat.div(batch, o2), Nat.mod(batch, o2)), Nat.mul(inner, columns)),
273    Matrix.read_offset(ro, inner, columns, Nat.mod(Nat.mod(position, Nat.mul(rows, columns)), columns), 0n))