Solved by jmandivarapu1 in post #8 
              
              
                You can flatten the model then the max of the numpy aray. Just pass your net to below function it will give you fattened numpy array of model and al it’s shapes. Then you can apply all numpy operations on that. 
#############################################################################
# Flatten…