Add batches to training episodes instead of one task

[tkarab]

• [x] Add extra dimention so the model can take numerous tasks in batches (model_assembly.py)
• [x] alter fsl_functions accordingly
• [x] add batch_size to task_generator
• [x] generate data in batches
• [x] Calculate loss across the whole batch (Taken care of by model.fit)