def second_written_projection

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

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

56def second_written_projection(-shape: Shape.Shape, -original: Storage.Storage, -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat}, +index: U32, +value: F32, +other: U32,
57  pair: Storage.Storage & Storage.Storage, -same: {pair == Storage.clone(original) : Storage.Storage & Storage.Storage}) ->
58  {Observation.clone_written_second_pair(shape, index, value, other, Core.clone_result(shape, original, valid, pair, same)) ==
59    Lifecycle.read_value(Storage.get(StorageLaws.outcome(Storage.Storage, pair), other)) : Maybe<&2, F32>}:
60  (first, second) = pair
61  untouched_value(shape, other, Core.write_flat(shape, Core.TTensor{first, Equal.trans(Nat, Lifecycle.logical_length(first), Lifecycle.logical_length(original), Shape.logical_numel(Shape.dimensions(shape)),
62      %Equal.sym(Storage.Storage & Storage.Storage, (first, second), Storage.clone(original), same) :
63        {Lifecycle.first_length(_) == Lifecycle.logical_length(original) : Nat}
64      Lifecycle.clone_first_length(original), valid)}, index, value), second,
65    Equal.trans(Nat, Lifecycle.logical_length(second), Lifecycle.logical_length(original), Shape.logical_numel(Shape.dimensions(shape)),
66      %Equal.sym(Storage.Storage & Storage.Storage, (first, second), Storage.clone(original), same) :
67        {Lifecycle.second_length(_) == Lifecycle.logical_length(original) : Nat}
68      Lifecycle.clone_second_length(original), valid))