def transpose_contract

bendtorch-core@0.1.0.0 · src/tensor/permute_values.bend:462

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis
import bendtorch-core@0.1.0.0/src/tensor/transpose_spec.bend as Transpose
import bendtorch-core@0.1.0.0/src/tensor/from_values_contract.bend as FromValues
import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Values
import bendtorch-core@0.1.0.0/src/tensor/transform_observation.bend as Observation
import bendtorch-core@0.1.0.0/src/tensor/permute_values_records.bend as Records

462def transpose_contract(+input: Shape.Shape, +values: List<&2, F32>,
463  counted: {List.length(&2, F32, values) == Shape.logical_numel(Shape.dimensions(input)) : Nat},
464  +a: U32, +b: U32, inside: {True{} == Transpose.valid(input, a, b) : Bool}) -> Records.TransposeContract<input, values, a, b>:
465  +counted = counted
466  +inside = inside
467  +axes = requested(input, a, b)
468  +valid = requested_valid(input, a, b, inside)
469  Records.TransposeContract{
470    FromValues.from_values_lift(input, values, counted,
471      result => {Observation.failure_code(input, Observation.Transpose{a, b}, result) == Some{0} : Maybe<&2, U32>},
472      fits => transpose_lift(input, values, a, b, inside, fits, pair => {Some{Observation.failure_code_pair(input, pair)} == Some{0} : Maybe<&2, U32>},
473        permuted_code(input, values, axes, valid, fits))),
474    FromValues.from_values_lift(input, values, counted,
475      result => {Observation.applied_dimensions(input, Observation.Transpose{a, b}, result) ==
476        Some{Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes))} : Maybe<&2, List<&2, U32>>},
477      fits => transpose_lift(input, values, a, b, inside, fits, pair => {Observation.applied_dimensions_pair(input, pair) ==
478          Some{Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes))} : Maybe<&2, List<&2, U32>>},
479        permuted_dimensions(input, values, axes, valid, fits))),
480    FromValues.from_values_lift(input, values, counted,
481      result => {Observation.applied_length(input, Observation.Transpose{a, b}, result) ==
482        Some{Shape.logical_numel(Shape.dimensions(input))} : Maybe<&2, Nat>},
483      fits => transpose_lift(input, values, a, b, inside, fits, pair => {Observation.applied_length_pair(input, pair) ==
484          Some{Shape.logical_numel(Shape.dimensions(input))} : Maybe<&2, Nat>},
485        permuted_length(input, values, axes, valid, fits))),
486    FromValues.from_values_lift(input, values, counted,
487      result => {Observation.applied_count(input, Observation.Transpose{a, b}, result) == Some{Shape.count(input)} : Maybe<&2, U32>},
488      fits => transpose_lift(input, values, a, b, inside, fits, pair => {Observation.applied_count_pair(input, pair) == Some{Shape.count(input)} : Maybe<&2, U32>},
489        permuted_count(input, values, axes, valid, fits))),
490    index => below =>
491      +below = below
492      FromValues.from_values_lift(input, values, counted,
493        result => {Observation.applied_value(input, Observation.Transpose{a, b}, index, result) ==
494          Values.listed(values, Values.permute_source(input, axes, index)) : Maybe<&2, F32>},
495        fits => transpose_lift(input, values, a, b, inside, fits, pair => {Observation.applied_value_pair(input, index, pair) ==
496            Values.listed(values, Values.permute_source(input, axes, index)) : Maybe<&2, F32>},
497          permuted_value(input, values, counted, axes, valid, index, below, fits))),
498    other => below =>
499      +below = below
500      FromValues.from_values_lift(input, values, counted,
501        result => {Observation.kept_value(input, Observation.Transpose{a, b}, other, result) ==
502          List.get(&2, F32, values, U32.to_nat(other)) : Maybe<&2, F32>},
503        fits => transpose_lift(input, values, a, b, inside, fits, pair => {Observation.kept_value_pair(input, other, pair) ==
504            List.get(&2, F32, values, U32.to_nat(other)) : Maybe<&2, F32>},
505          permuted_kept(input, values, counted, axes, valid, other, below, fits)))}