def permute_contract

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

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/axis_spec.bend as Axis
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

391def permute_contract(+input: Shape.Shape, +values: List<&2, F32>,
392  counted: {List.length(&2, F32, values) == Shape.logical_numel(Shape.dimensions(input)) : Nat},
393  +axes: List<&2, U32>, valid: {True{} == Layout.bijective(input, axes) : Bool}) -> Records.PermuteContract<input, values, axes>:
394  +counted = counted
395  +valid = valid
396  Records.PermuteContract{
397    FromValues.from_values_lift(input, values, counted,
398      result => {Observation.failure_code(input, Observation.Permute{axes}, result) == Some{0} : Maybe<&2, U32>},
399      fits => permuted_code(input, values, axes, valid, fits)),
400    FromValues.from_values_lift(input, values, counted,
401      result => {Observation.applied_dimensions(input, Observation.Permute{axes}, result) ==
402        Some{Axis.order(U32, Shape.dimensions(input), Shape.naturals(axes))} : Maybe<&2, List<&2, U32>>},
403      fits => permuted_dimensions(input, values, axes, valid, fits)),
404    FromValues.from_values_lift(input, values, counted,
405      result => {Observation.applied_length(input, Observation.Permute{axes}, result) ==
406        Some{Shape.logical_numel(Shape.dimensions(input))} : Maybe<&2, Nat>},
407      fits => permuted_length(input, values, axes, valid, fits)),
408    FromValues.from_values_lift(input, values, counted,
409      result => {Observation.applied_count(input, Observation.Permute{axes}, result) == Some{Shape.count(input)} : Maybe<&2, U32>},
410      fits => permuted_count(input, values, axes, valid, fits)),
411    index => inside =>
412      +inside = inside
413      FromValues.from_values_lift(input, values, counted,
414        result => {Observation.applied_value(input, Observation.Permute{axes}, index, result) ==
415          Values.listed(values, Values.permute_source(input, axes, index)) : Maybe<&2, F32>},
416        fits => permuted_value(input, values, counted, axes, valid, index, inside, fits)),
417    other => inside =>
418      +inside = inside
419      FromValues.from_values_lift(input, values, counted,
420        result => {Observation.kept_value(input, Observation.Permute{axes}, other, result) ==
421          List.get(&2, F32, values, U32.to_nat(other)) : Maybe<&2, F32>},
422        fits => permuted_kept(input, values, counted, axes, valid, other, inside, fits))}