MACs & parameters calculation for neural network, PyTorch

Simple to use, easy to extend.

See

python macs_profiling.py

for usage.