TensorBackward.batched_matmul: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
Lines 1285-1291 of bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&start_line=1285&end_line=1291

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

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

Uses (7):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Result
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&name=Binary
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&name=Error
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&name=Tape
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&name=binary