DeepCopyOp [id A] 9
 └─ w_simplex__ [id B]
DeepCopyOp [id C] 3
 └─ x_coord_ordered__ [id D]
DeepCopyOp [id E] 1
 └─ y_coord [id F]
DeepCopyOp [id G] 7
 └─ sigma_log__ [id H]
Softmax{axis=0} [id I] 'w' 23
 └─ Sub [id J] 22
    ├─ Join [id K] 16
    │  ├─ 0 [id L]
    │  ├─ w_simplex__ [id B]
    │  └─ Neg [id M] 14
    │     └─ ExpandDims{axis=0} [id N] 12
    │        └─ Sum{axes=None} [id O] 8
    │           └─ w_simplex__ [id B]
    └─ ExpandDims{axis=0} [id P] 20
       └─ Max{axis=0} [id Q] 'max' 18
          └─ Join [id K] 16
             └─ ···
CumOp{-1, add} [id R] 'x_coord' 15
 └─ SetSubtensor{start:} [id S] 13
    ├─ SetSubtensor{i} [id T] 11
    │  ├─ Alloc [id U] 5
    │  │  ├─ 0.0 [id V]
    │  │  └─ 5 [id W]
    │  ├─ Subtensor{i} [id X] 4
    │  │  ├─ x_coord_ordered__ [id D]
    │  │  └─ 0 [id Y]
    │  └─ 0 [id Y]
    ├─ Exp [id Z] 10
    │  └─ Subtensor{start:} [id BA] 2
    │     ├─ x_coord_ordered__ [id D]
    │     └─ 1 [id BB]
    └─ 1 [id BB]
Exp [id BC] 'sigma' 6
 └─ sigma_log__ [id H]
Transpose{axes=[1, 0]} [id BD] 'centroids' 21
 └─ Join [id BE] 19
    ├─ 0 [id L]
    ├─ ExpandDims{axis=0} [id BF] 17
    │  └─ CumOp{-1, add} [id R] 'x_coord' 15
    │     └─ ···
    └─ ExpandDims{axis=0} [id BG] 0
       └─ y_coord [id F]