~/jendhub

def catg_b

bendtorch@1.3.0.0 · tensor.bend:615

import bendtorch@1.3.0.0/tensor.bend as Tensor
import Base

615def catg_b(+a: List<&2, Nat>, +b: List<&2, Nat>, +d: Nat, r: P2) -> CatG<a, b, cat_shape(a, b, d)>:
616  match r:
617    case P2{dz, ga}:
618      catg_c(a, b, d, ga, gather_l(numel(b), 0, cat_map_b(a, b, d), Array.get(F32, dz, imap(cat_map_b(a, b, d), 0)), new(numel(b))))

Uses (13)

Used by (1)