def Vec4.mask

bend-tensors@0.0.0.2 · bend_tensors.bend:323

import Base

323# elements at index >= k set to 0.0
324def Vec4.mask(k: Nat, v: Vec4) -> Vec4:
325  match k v:
326    case 0n Vec4{x0, x1, x2, x3}:
327      Vec4{0.0, 0.0, 0.0, 0.0}
328    case 1n Vec4{x0, x1, x2, x3}:
329      Vec4{x0, 0.0, 0.0, 0.0}
330    case 2n Vec4{x0, x1, x2, x3}:
331      Vec4{x0, x1, 0.0, 0.0}
332    case 3n Vec4{x0, x1, x2, x3}:
333      Vec4{x0, x1, x2, 0.0}
334    case _ v2:
335      v2