base.py

We have three classes in this file:

• Operation is base class for operations between tensors.
• Module is base class for neural network parts. This name is just like pytorch.
• Optimizer is base class for Gradient Descent optimizers.