def one_hot_row
import mlbend-preprocess@0.1.0.0/main.bend as MlbendPreprocessimport Base
255# One row of a one-hot matrix: a zero in every column but the label's.
256def one_hot_row(+label: Nat, k: Nat) -> List<&2, F32>:
257 replace_at(List.replicate(F32, k, 0.0), label, 1.0)