def shaped

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

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

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

154def shaped(+left: Operand, +right: Operand) -> Result<&2, &2, Matrix.Error, Plan>:
155  Operand{Batch{+lr, +l1, +l2}, +rows, +inner, +lv} = left
156  Operand{Batch{+rr, +r1, +r2}, +columns, right_inner, +rv} = right
157  batched(larger(lr, rr), rows, inner, columns, l1, l2, r1, r2, lv, rv, axis(l1, r1), axis(l2, r2))

Uses (8)

Used by (1)

bendtorch-core@0.1.0.0/src/tensor/batched_matmul:compatible