def max_balanced
bendtorch-core@0.1.0.0 · src/autograd/loss.bend:133
import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Loss
import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel
import bendtorch-core@0.1.0.0/src/autograd/reduce.bend as Reduce
133def max_balanced(+depth: Nat, +splits: Nat, +columns: Nat, +count: Nat, +base: Nat, array: Array<F32>) ->
134 Model.Balanced(F32, depth, max_build(depth, splits, columns, count, base, array)):
135 Parallel.balanced(~Width, ~Width, ~Array<F32>, ~Unit, ~width, ~Reduce.split, ~max_cell, Width{columns}, depth, splits, count, base,
136 (array, Unit{}))