type Layout

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

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

194# What every cell of a build reads: the orders and extents of the product
195# (as in `Matrix.Layout`), and each operand's padded batch extents with the
196# output's second batch extent, which splits a batch number into its two
197# indices.
198type Layout is Data:
199  Layout{left: Matrix.Order, right: Matrix.Order, inner: Nat, rows: Nat, columns: Nat,
200    l1: Nat, l2: Nat, r1: Nat, r2: Nat, o2: Nat}