def unflatten

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

import bendtorch-core@0.1.0.0/src/shape/core.bend as Core

import Base

159def unflatten(+dims: List<&2, Nat>, index: Fin(numel(dims))) -> Coordinate(dims):
160  match dims:
161    case Nil{}:
162      Unit{}
163    case +head <> +tail:
164      unflatten.step(head, tail, rest => unflatten(tail, rest), Fin.unpack(head, numel(tail), index))

Uses (8)

Used by (22)