I got it but how can I decide the shape of hidden dimension, I am confused
" suppose I have 2d input with shape (1,2) and hidden size 16 and output with shape (1,1), I can say hidden shape is  (2,16) but for 3d if input shape is (1,1,2), [batch_size, seq_len, features]  & output (1,1,1) how can i calculate the shape of hidden layer