def castp_f2

bendtorch@1.3.0.0 · dtype.bend:637

import Base
import bendtorch@1.3.0.0/main.bend as T
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

637def castp_f2(-X: Type, -Y: Type, -S: Type, -r: Nat, -c: Nat, w: TA.Mat<r, c>, f: T.Fwd<TA.Mat<r, c>, Y, S>) -> T.Fwd<TA.Mat<r, c>, Y, S>:
638  match f:
639    case T.Fwd{wc, y, s}:
640      T.Fwd{w, y, s}