def rope_go2

bendtorch@1.3.0.0 · main.bend:2324

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

2324def rope_go2(+n: Nat, +w: Nat, +fwd: Bool, +t: Nat, +h: Nat, +half: Nat, x: Array<F32>, tb: CS) -> TA.Mat<n, w>:
2325  match tb:
2326    case CS{cs, sn}:
2327      rope_fin(n, w, rope_e(Nat.sub(Nat.mul(n, Nat.mul(h, half)), 1n), 0, fwd, U32.from_nat(Nat.mul(h, half)), U32.from_nat(half), U32.from_nat(w), U32.from_nat(t), RS3{x, cs, sn}))