def axis

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

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

88def axis(+l: U32, +r: U32) -> Result<&2, &2, Matrix.Error, U32>:
89  stretched(l, r, Nat.is_eq(U32.to_nat(l), U32.to_nat(r)), Nat.is_eq(U32.to_nat(l), 1n), Nat.is_eq(U32.to_nat(r), 1n))