def length_result

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

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/tensor/lifecycle.bend as Lifecycle

344def length_result(-shape: Shape.Shape, -original: Storage.Storage,
345  -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat},
346  pair: Storage.Storage & Nat, -same: {pair == Storage.length(original) : Storage.Storage & Nat}) -> Tensor<shape> & Nat:
347  (storage, n) = pair
348  (TTensor{storage, Equal.trans(Nat, Lifecycle.logical_length(storage), Lifecycle.logical_length(original),
349    Shape.logical_numel(Shape.dimensions(shape)),
350    %Equal.sym(Storage.Storage & Nat, (storage, n), Storage.length(original), same) :
351      {Lifecycle.length_owner(_) == Lifecycle.logical_length(original) : Nat}
352    Lifecycle.length_preserved(original), valid)}, n)