Thank you for your time.
The input is basically a 500x42x51. I want the Neural Network to basically take every 42x51 and classify that input as 0 or 1(output). Does it make sense?
Here, I have randomly created a 500x42x51 matrix using torch. I have the actual input data of the same shape.
The output is of shape 500x1 consisting of either 0’s or 1’s. For each 42x51 input there is an output 0 or 1.