def first_value_projection

bendtorch-core@0.1.0.0 · src/tensor/clone_contract.bend:14

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/main.bend as Storage
import bendtorch-core@0.1.0.0/src/storage/LAWS.bend as StorageLaws
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/lifecycle.bend as Lifecycle
import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Observation

14def first_value_projection(-shape: Shape.Shape, -original: Storage.Storage, -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat}, +other: U32,
15  pair: Storage.Storage & Storage.Storage, -same: {pair == Storage.clone(original) : Storage.Storage & Storage.Storage}) ->
16  {Observation.clone_first_value_pair(shape, other, Core.clone_result(shape, original, valid, pair, same)) ==
17    Lifecycle.read_value(Storage.get(StorageLaws.owner(Storage.Storage, pair), other)) : Maybe<&2, F32>}:
18  (first, second) = pair
19  Core.read_result_value(shape, first,
20    Equal.trans(Nat, Lifecycle.logical_length(first), Lifecycle.logical_length(original), Shape.logical_numel(Shape.dimensions(shape)),
21      %Equal.sym(Storage.Storage & Storage.Storage, (first, second), Storage.clone(original), same) :
22        {Lifecycle.first_length(_) == Lifecycle.logical_length(original) : Nat}
23      Lifecycle.clone_first_length(original), valid),
24    other, Storage.get(first, other), {==})