def right_below

bendtorch-core@0.1.0.0 · src/tensor/matrix_index_proof.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/shape/nat_proofs.bend as Nats

56def right_below(+inner: Nat, +columns: Nat, +column: Nat, column_below: {Nat.is_lt(column, columns) == True{} : Bool}, +right: Shape.Shape,
57  right_extents: {Shape.extents(right) == [inner, columns] : List<&2, Nat>}, +k: Nat, below: {Nat.is_lt(k, inner) == True{} : Bool}) ->
58  {Nat.is_lt(Nat.add(Nat.mul(k, columns), column), Shape.logical_numel(Shape.dimensions(right))) == True{} : Bool}:
59  %Equal.sym(List<&2, Nat>, Shape.extents(right), [inner, columns], right_extents) :
60    {Nat.is_lt(Nat.add(Nat.mul(k, columns), column), Shape.numel(_)) == True{} : Bool}
61  %Equal.sym(Nat, Nat.mul(columns, 1n), columns, Nats.mul_one_right(columns)) :
62    {Nat.is_lt(Nat.add(Nat.mul(k, columns), column), Nat.mul(inner, _)) == True{} : Bool}
63  lt_mul_add(k, inner, column, columns, below, column_below)