~/jendhub

def axis_meaning

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_PROOF.bend:639

import bendtorch-core@0.1.0.0/src/tensor/batched_matmul_PROOF.bend as BatchedMatmulPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as Batched

639# One batch axis of the plan has the extent PyTorch's broadcast gives it.
640def axis_meaning(+l: U32, +r: U32) -> AxisMeaning(l, r, Batched.axis(l, r)):
641  axis_same(l, r, Shape.bool_case(Nat.is_eq(U32.to_nat(l), U32.to_nat(r))))

Uses (7)

Used by (1)