其实看RNN源码即可知晓

首先是hx的维度通过以下源码可知道
hx的维度是

(self.num_layers * num_directions,
                                batch_sizes[0], self.hidden_size)

但是batch_size[0]要参考你设置的参数
如下源码可知

max_batch_size = input.size(0) if self