search

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
MIT License
1.79k stars 266 forks source link

Fold parallel #70

Open indigoviolet opened 4 years ago

indigoviolet commented 4 years ago

image

I can't seem to fold this into one module with something like

hl.transforms.Fold('((Shape > Gather > Unsqueeze) | (Shape > Gather > Unsqueeze)) > Concat', 'UnsqueezeThing')

Should this work? What's the correct expression?

Thank you for sharing this library, it's pretty great!