def checked_keeps
bendtorch-core@0.1.0.0 · src/autograd/sum_dims_PROOF.bend:79
import bendtorch-core@0.1.0.0/src/autograd/sum_dims_PROOF.bend as SumDimsPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
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
79def checked_keeps(+axes: List<&2, U32>, +dimensions: List<&2, U32>, checked: Maybe<&2, Layout.Error>) ->
80 Laws.PlanKeeps(axes, dimensions, SumDims.plan_checked(axes, dimensions, checked)):
81 match checked:
82 case Some{error}:
83 Unit{}
84 case None{}:
85 built_keeps(axes, dimensions, Shape.construct(SumDims.kept_dimensions(axes, dimensions)))