Hello,
I’m new to NNs and pytorch!
I have some data being provided to my model via a DataLoader with the batch size set to 8. The shape of my input is (8, 15, 50).
I want to make a simple FFNN to classify my data, there are 50 different classes for my data.
So, I’m trying to do Linear, ReLU, Linear, ReLU, Linear (two hidden layers and then a final Linear that outputs a 50-dimensional vector).
Given that the initial input is (batch_size, 15, 50), what do I need to do to the shape so that the initial Linear will work on it? I thought it should be 15*50 on the first layer? But I think I need to reshape the (8, 15, 50) input?
Thank you.