def right_below

bendtorch-core@0.1.0.0 · src/tensor/contracts/dot_proof.bend:75

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/shape/nat_proofs.bend as Nats
import bendtorch-core@0.1.0.0/src/tensor/contracts/dot.bend as Dot

75def right_below(+inner: Nat, +columns: Nat, +row: Nat, +column: Nat,
76  column_below: {Nat.is_lt(column, columns) == True{} : Bool}, +right: Shape.Shape,
77  right_extents: {Shape.extents(right) == [inner, columns] : List<&2, Nat>}, +k: Nat, below: {Nat.is_lt(k, inner) == True{} : Bool}) ->
78  {Nat.is_lt(Dot.right_index(Dot.Cell{row, column, inner, columns}, k), Shape.logical_numel(Shape.dimensions(right))) == True{} : Bool}:
79  %Equal.sym(List<&2, Nat>, Shape.extents(right), [inner, columns], right_extents) :
80    {Nat.is_lt(Nat.add(Nat.mul(k, columns), column), Shape.numel(_)) == True{} : Bool}
81  %Equal.sym(Nat, Nat.mul(columns, 1n), columns, Nats.mul_one_right(columns)) :
82    {Nat.is_lt(Nat.add(Nat.mul(k, columns), column), Nat.mul(inner, _)) == True{} : Bool}
83  lt_mul_add(k, inner, column, columns, below, column_below)