Model summary in PyTorch similar to `model.summary()` in Keras
