def dup_fin

bendtorch-fast@1.3.0.0 · main.bend:223

import bendtorch-fast@1.3.0.0/main.bend as BendtorchFast
import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA
import bend-ml-tensor-fork@0.1.5.0/main.bend as TF
import bendtorch@1.3.0.0/main.bend as T

223def dup_fin(-r: Nat, -c: Nat, q: TF.B2<F32, F32>) -> T.Two<TA.Mat<r, c>, TA.Mat<r, c>>:
224  match q:
225    case TF.B2{a, b}:
226      T.Two{TA.Mat{a}, TA.Mat{b}}

Uses (5)

Used by (1)