I am new to machine learning and trying to train a model which predicts what a matrix would look like based on series of input matrices. The scenario is as follows: We have