PyTorch LSTM,batch_first=True对初始化h0和c0的影响

batch_first=True会对LSTM的输入输出的维度顺序有影响,但是对初始化h0和c0的维度顺序没有影响,也就是说,不管batch_first=True还是False,h0和c0的维度顺序都是:

 

 

关于LSTM的输入输出,可参考这篇博客

 

posted @ 2020-09-09 09:45  Picassooo  阅读(2012)  评论(0编辑  收藏  举报