def owner_value_projection

bendtorch-core@0.1.0.0 · src/tensor/metadata_contract.bend:13

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

13def owner_value_projection(-shape: Shape.Shape, -original: Storage.Storage, -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat}, +other: U32,
14  pair: Storage.Storage & Nat, -same: {pair == Storage.length(original) : Storage.Storage & Nat}) ->
15  {Observation.length_owner_value_pair(shape, other, Core.length_result(shape, original, valid, pair, same)) ==
16    Lifecycle.read_value(Storage.get(StorageLaws.owner(Nat, pair), other)) : Maybe<&2, F32>}:
17  (storage, number) = pair
18  Core.read_result_value(shape, storage,
19    Equal.trans(Nat, Lifecycle.logical_length(storage), Lifecycle.logical_length(original), Shape.logical_numel(Shape.dimensions(shape)),
20      %Equal.sym(Storage.Storage & Nat, (storage, number), Storage.length(original), same) :
21        {Lifecycle.length_owner(_) == Lifecycle.logical_length(original) : Nat}
22      Lifecycle.length_preserved(original), valid),
23    other, Storage.get(storage, other), {==})