def VecT.shift

bend-tensors@0.0.0.2 · bend_tensors.bend:549

import Base

549# k added to every element
550def VecT.shift(+n: Nat, +k: F32, v: VecT(n)) -> VecT(n):
551  match n:
552    case 0n:
553      VecL{x} = v
554      Vec4{x0, x1, x2, x3} = x
555      VecL{Vec4{(x0 + k : F32), (x1 + k : F32), (x2 + k : F32), (x3 + k : F32)}}
556    case 1n+p:
557      VecN{l, r} = v
558      a b = VecT.shift(p, k, l) VecT.shift(p, k, r)
559      VecN{a, b}