def single_column

bendtorch-core@0.1.0.0 · src/autograd/loss.bend:61

import Base

61# Whether [rows, columns] logits have one column, one class.
62def single_column(+dimensions: List<&2, U32>) -> Bool:
63  match dimensions:
64    case [+rows, +columns]:
65      U32.is_eq(columns, 1)
66    case _:
67      False{}