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?