def left_from

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_PROOF.bend:378

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

378# The first left read of a cell, from its batch number `batch` and its
379# offset `within` inside its matrix.
380def left_from(+lo: Matrix.Order, +inner: Nat, +rows: Nat, +columns: Nat, +l1: Nat, +l2: Nat, +o2: Nat, +batch: Nat, +within: Nat) -> Nat:
381  Nat.add(Nat.mul(Batched.batch_number(l1, l2, Nat.div(batch, o2), Nat.mod(batch, o2)), Nat.mul(rows, inner)),
382    Matrix.read_offset(lo, inner, rows, Nat.div(within, columns), 0n))