def cat_b

bendtorch@1.3.0.0 · tensor.bend:595

import bendtorch@1.3.0.0/tensor.bend as Tensor

import Base

595def cat_b(+a: List<&2, Nat>, +b: List<&2, Nat>, +d: Nat, rb: Array<F32>, pa: P2) -> T3<a, b, cat_shape(a, b, d)>:
596  match pa:
597    case P2{ax, o}:
598      t3_cat(a, b, cat_shape(a, b, d), ax, place_l(numel(b), 0, cat_map_b(a, b, d), Array.get(F32, rb, 0), o))

Uses (12)

Used by (1)

bendtorch@1.3.0.0/tensor:cat