def bmm_l

bendtorch@1.3.0.0 · tensor.bend:674

import Base

674# out[j] for cnt positions from j
675def bmm_l(cnt: Nat, +j: U32, +g: Geo, x: Array<F32>, y: Array<F32>, o: Array<F32>) -> P3:
676  match cnt:
677    case 0n:
678      P3{x, y, o}
679    case 1n+p:
680      Dot.then(P3, dot_l(geo_k(g), x_off(g, j), geo_xt(g), y_off(g, j), geo_yt(g), Array.get(F32, x, x_off(g, j)), Array.get(F32, y, y_off(g, j)), 0.0),
681        x2 => y2 => v => bmm_l(p, (j + 1 : U32), g, x2, y2, Array.set(F32, o, j, v)))