在RNN的前向函数中使用了许多Pytorch例子
self.rnn.flatten_parameters()
Resets parameter data pointer so that they can use faster code paths
重置参数数据指针,以便使用更快的代码路径
我认为它只是把你所有的weight压缩到一个连续的内存chuck中
为了提高内存的利用率和效率,调用flatten_parameters让parameter的数据存放成contiguous chunk(连续的块)。类似我们调用tensor.contiguous
pytorch - What does flatten_parameters() do? - Stack Overflow