def step_value

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

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

788def step_value(-input: Shape.Shape, -axes: List<&2, U32>, -plan: Layout.Plan<input, axes>, -index: U32,
789  -before_source: Core.Tensor<input>, -before_target: Core.Tensor<Layout.output(input, axes, plan)>,
790  -after_source: Core.Tensor<input>, -after_target: Core.Tensor<Layout.output(input, axes, plan)>,
791  meaning: Transforms.StepMeaning(input, axes, plan, index, before_source, before_target, (after_source, Done{after_target}))) ->
792  {Core.checked_observe(Layout.output(input, axes, plan), after_target, index) ==
793    Transforms.expected_output(input, axes, plan, before_source, index) : Maybe<&2, F32>}:
794  (kept, rest) = meaning
795  (checked, values) = rest
796  (observed, frame) = values
797  observed