But then I’d have to train using stochastic gradient descent, right? By showing one training example at a time? What I want to achieve is: to train on batches and predict a single example.
I think that key to my concern is a lack of proper understanding of what init_hidden does, as it includes in return dimensions self.batch_size, which are not needed for me at a prediction stage.