law output_dimensions

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix

201law output_dimensions:
202  for -left: Shape.Shape
203  for -right: Shape.Shape
204  for +output: Shape.Shape
205  for rows: U32
206  for inner: U32
207  for right_inner: U32
208  for columns: U32
209  for le: {Shape.dimensions(left) == [rows, inner] : List<&2, U32>}
210  for re: {Shape.dimensions(right) == [right_inner, columns] : List<&2, U32>}
211  for oe: {Shape.dimensions(output) == [rows, columns] : List<&2, U32>}
212  for compatible: {U32.to_nat(inner) == U32.to_nat(right_inner) : Nat}
213  {Shape.dimensions(Matrix.output_shape(left, right, Matrix.Plan{output, rows, inner, right_inner, columns, le, re, oe, compatible})) ==
214    [rows, columns] : List<&2, U32>}