def scatter_balanced
bendtorch-core@0.1.0.0 · src/tensor/slice.bend:290
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice
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/tensor/elementwise.bend as Elementwise
290def scatter_balanced(+depth: Nat, +splits: Nat, +plan: Plan, +count: Nat, +base: Nat, g: Array<F32>) ->
291 Model.Balanced(F32, depth, scatter_build(depth, splits, plan, count, base, g)):
292 Parallel.balanced(~Plan, ~Plan, ~Array<F32>, ~Array<F32>, ~same, ~Elementwise.split, ~scatter_cell, plan, depth, splits, count, base,
293 (g, Array.new(F32, 0n, 0.0)))