def rope_fin

bendtorch@1.3.0.0 · main.bend:2319

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

2319def rope_fin(-n: Nat, -w: Nat, st: RS3) -> TA.Mat<n, w>:
2320  match st:
2321    case RS3{x, cs, sn}:
2322      TA.Mat{x}