May I get the very first and last tensor shape?

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

Open hunkim opened 4 years ago

hunkim commented 4 years ago

This is really wonderful!

This is one example. Is it possible to get the tensorshape of the very first (input) and final output? It would be very useful.