def batch_number

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

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

import Base

246def batch_number(+first: Nat, +second: Nat, +b1: Nat, +b2: Nat) -> Nat:
247  Nat.add(Nat.mul(stretch(b1, Nat.is_eq(first, 1n)), second), stretch(b2, Nat.is_eq(second, 1n)))

Uses (5)

Used by (7)