I use resnet to predict EEG response to image data. And the input data is a matrix x_i with size of 1 * 3 * 500 * 500. Then the output is a vector y_i with size of 1 * 1 * 17 * 100. Let’s call the output EEG response. there is a loop like the following:

for i in range(number of input data):

y_i = model(x_i)

if max_i=200, I would to stack all the y_i to create a matrix with size of 200 * 1 * 17 * 100. How to realize that?