def built_keeps

bendtorch-core@0.1.0.0 · src/autograd/sum_dims_PROOF.bend:69

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/autograd/sum_dims.bend as SumDims
import bendtorch-core@0.1.0.0/src/autograd/sum_dims_LAWS.bend as Laws

69def built_keeps(+axes: List<&2, U32>, +dimensions: List<&2, U32>,
70  built: Shape.ConstructResult<SumDims.kept_dimensions(axes, dimensions)>) ->
71  Laws.PlanKeeps(axes, dimensions,
72    SumDims.plan_built(SumDims.kept_dimensions(axes, dimensions), SumDims.dropped_dimensions(axes, dimensions), built)):
73  match built:
74    case Shape.SError{error}:
75      Unit{}
76    case Shape.SBuilt{+kept, preserved}:
77      (preserved, {==})