def project_indices

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

import Base

202def project_indices(+dims: List<&2, Nat>, +values: List<&2, Nat>) -> List<&2, Nat>:
203  broadcast_indices(dims, List.drop(&2, Nat, values, Nat.sub(List.length(&2, Nat, values), List.length(&2, Nat, dims))))