def cat_bwd

import bendtorch@1.3.0.0/tensor.bend as Tensor

import Base

620def cat_bwd(+a: List<&2, Nat>, +b: List<&2, Nat>, +d: Nat, dz: Tensor<cat_shape(a, b, d)>) -> CatG<a, b, cat_shape(a, b, d)>:
621  match dz:
622    case Tensor{dd}:
623      catg_b(a, b, d, gather_l(numel(a), 0, cat_map_a(a, b, d), Array.get(F32, dd, imap(cat_map_a(a, b, d), 0)), new(numel(a))))

Uses (13)

List
Nat
F32
Array.get
bendtorch@1.3.0.0/tensor:Tensor
…

Used by (0)

No other definition uses this definition.