Hmm! maybe! I’ve been eyeballing that line for two days wondering why I can’t find anyone else using something like that.
The nn.LSTM adds an undesired “batch_size” dimension that the nn.LSTMCell will reject. So I’ve got to squeeze it somehow… maybe in that process I’ve stopped backprop. But how else would I accomplish the squeeze?