Thank you very much for fast reply~
I think both idea looks okay but the first one is muc more close. From your suggestion I understood that I should return 3d arrays and seperate on the training loop… I am sorry I couldn`t understand clearly how can I iterate them in training loop to 2D arrays?