BendTensors.Blk.mask_cols: def of bend-tensors@0.0.0.2
Import: import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
Lines 399-423 of bend-tensors@0.0.0.2/bend_tensors.bend: https://jend.leeeet.dev/read.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&start_line=399&end_line=423

import Base

# columns at index >= k set to 0.0
def Blk.mask_cols(k: Nat, m: Blk) -> Blk:
  match k m:
    case 0n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
      Blk{0.0, 0.0, 0.0, 0.0,
      0.0, 0.0, 0.0, 0.0,
      0.0, 0.0, 0.0, 0.0,
      0.0, 0.0, 0.0, 0.0}
    case 1n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
      Blk{a00, 0.0, 0.0, 0.0,
      a10, 0.0, 0.0, 0.0,
      a20, 0.0, 0.0, 0.0,
      a30, 0.0, 0.0, 0.0}
    case 2n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
      Blk{a00, a01, 0.0, 0.0,
      a10, a11, 0.0, 0.0,
      a20, a21, 0.0, 0.0,
      a30, a31, 0.0, 0.0}
    case 3n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
      Blk{a00, a01, a02, 0.0,
      a10, a11, a12, 0.0,
      a20, a21, a22, 0.0,
      a30, a31, a32, 0.0}
    case _ m2:
      m2

Uses (2):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=Blk