Consider I have a input data
X = torch.Tensor(sample_size,dim1,dim2)
For each data point, its a matrix form with dim1xdim2
Now when I feed into neural network with
nn.Linear(input_Dim,hidden_dim)
I have
output = input.matmul(weight.t())
RuntimeError: size mismatch,

This makes sense because forward only does operation of matrix multiply with a vector
How should I modify my neural net work?