Hey, one more question. The code above works in case I want to get the representation of the original data in the middle layers. What if I want to know the weights in the middle layers?
For e.g., as per @ptrblck’s code, can I get the 50x20 and 20x5 matrices containing the weights?