def batched_matmul

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1285

import Base
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as Batched

1285# Records torch.matmul(a, b): batched products and 1-D promotion
1286# (Batched.matmul). Two rank-2 operands record exactly what `matmul` records.
1287# A rank-0 operand is refused with RankMismatch (105), unequal inner extents
1288# with InnerDimensionMismatch (106), batch dimensions that do not broadcast
1289# with IncompatibleBroadcast (6); the tape comes back unchanged.
1290def batched_matmul(tape: Tape, +a: Nat, +b: Nat) -> Tape & Result<&2, &2, Error, Nat>:
1291  binary(tape, BatchedProduct{}, a, b)