def Flat.mask_causal

bend-tensors@0.0.0.2 · bend_tensors.bend:1889

import Base

1889def Flat.mask_causal(+rows: Nat, +cols: Nat, m: Flat<rows, cols>) -> Flat<rows, cols>:
1890  FlatS{a} = m
1891  FlatS{Flat.mask_causal.go(Flat.count(rows, cols), 0, U32.from_nat(cols), a)}