def dot_at

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

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

275# Cell `position`: all `inner` terms, from +0.0.
276def dot_at(+layout: Layout, +position: Nat, la: Array<F32>, ra: Array<F32>) -> F32 & (Array<F32> & Array<F32>):
277  Layout{+lo, +ro, +inner, +rows, +columns, l1, l2, r1, r2, o2} = layout
278  run_from(inner, Matrix.stride(lo, rows), Matrix.stride(ro, columns), left_start(layout, position), right_start(layout, position), la, ra)