Python Model Summary Pytorch Vs Tensorflow