~/jendhub

def broadcast_batch

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_LAWS.bend:343

import bendtorch-core@0.1.0.0/src/tensor/batched_matmul_LAWS.bend as BatchedMatmulLAWS
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

343# Two batches broadcast right-aligned: the shorter one is padded with ones
344# on the left (`Shape.pad`), then each axis broadcasts.
345def broadcast_batch(+left: List<&2, Nat>, +right: List<&2, Nat>) -> List<&2, Nat>:
346  zip_extents(Shape.pad(Nat.sub(List.length(&2, Nat, right), List.length(&2, Nat, left)), left),
347    Shape.pad(Nat.sub(List.length(&2, Nat, left), List.length(&2, Nat, right)), right))

Uses (6)

Used by (2)