def natural_step_lift

bendtorch-core@0.1.0.0 · src/tensor/transform_PROOF.bend:651

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/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms
import bendtorch-core@0.1.0.0/src/tensor/lifecycle.bend as Lifecycle

651def natural_step_lift(+input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +cursor: Nat, +index: U32,
652  -source: Core.Tensor<input>, -target: Core.Tensor<Layout.output(input, axes, plan)>,
653  -candidate: {index == U32.from_nat(cursor) : U32},
654  -conversion: {True{} == Nat.is_eq(U32.to_nat(index), cursor) : Bool},
655  same: {Transforms.natural_decided(input, axes, plan, cursor, index, source, target, candidate, Shape.BTrue{conversion}) ==
656    Transforms.natural_step(input, axes, plan, cursor, source, target) : Transforms.NaturalCopy<input, axes, plan, cursor>},
657  certificate: Transforms.StepCertificate<input, axes, plan, index, source, target>) ->
658  Transforms.NaturalCertificate<input, axes, plan, cursor, source, target>:
659  Transforms.StepCertificate{pair, exact, meaning} = certificate
660  Transforms.NaturalCertificate{Transforms.NaturalCopied{pair, index, Lifecycle.natural_equal(U32.to_nat(index), cursor, conversion), candidate},
661    Equal.trans(Transforms.NaturalCopy<input, axes, plan, cursor>,
662      Transforms.NaturalCopied{pair, index, Lifecycle.natural_equal(U32.to_nat(index), cursor, conversion), candidate},
663      Transforms.natural_decided(input, axes, plan, cursor, index, source, target, candidate, Shape.BTrue{conversion}),
664      Transforms.natural_step(input, axes, plan, cursor, source, target),
665      Equal.cong(Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>>,
666        Transforms.NaturalCopy<input, axes, plan, cursor>,
667        result => Transforms.natural_result(input, axes, plan, cursor, index, Lifecycle.natural_equal(U32.to_nat(index), cursor, conversion), candidate, result),
668        pair, Transforms.copy_step(input, axes, plan, source, target, index), exact), same), meaning}