~/jendhub

def Flat.mask_causal.go

bend-tensors@0.0.0.2 · bend_tensors.bend:1882

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1882def Flat.mask_causal.go(n: Nat, +idx: U32, +colsu: U32, a: Array<F32>) -> Array<F32>:
1883  match n:
1884    case 0n:
1885      a
1886    case 1n+p:
1887      Flat.mask_causal.go(p, (idx + 1 : U32), colsu, Flat.mask_causal.step(idx, U32.div(idx, colsu), U32.mod(idx, colsu), Array.get(F32, a, idx)))

Uses (9)

Used by (1)