Batching

**Is your feature request related to a problem? Please describe.**
Add the possibility of using the batching within a PINN train.
 
**Describe the solution you'd like**
A new argument in the constructor to select the batch size and using it in the optimization cycle.

**Additional context**
Mandatory for long train, especially using GPU