def broadcast_indices

bendtorch-core@0.1.0.0 · src/shape/core.bend:193

import Base

193def broadcast_indices(dims: List<&2, Nat>, values: List<&2, Nat>) -> List<&2, Nat>:
194  match dims values:
195    case Nil{} _:
196      Nil{}
197    case dh <> dt Nil{}:
198      Nil{}
199    case dh <> dt vh <> vt:
200      broadcast_index(dh, vh) <> broadcast_indices(dt, vt)