type Plan

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

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

120# Everything a product and its two gradients need: the output shape, the
121# output's batch dimensions, the matrix extents, both operands' padded
122# batches and the output's, and which operands are vectors.
123type Plan is Data:
124  Plan{output: Shape.Shape, batch: List<&2, U32>, rows: Nat, inner: Nat, columns: Nat,
125    l1: Nat, l2: Nat, r1: Nat, r2: Nat, o1: Nat, o2: Nat, left_vector: Bool, right_vector: Bool}