batch_first=True会对LSTM的输入输出的维度顺序有影响,但是对初始化h0和c0的维度顺序没有影响,也就是说,不管batch_first=True还是False,h0和c0的维度顺序都是:
关于LSTM的输入输出,可参考这篇博客。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:PyTorch LSTM,batch_first=True对初始化h0和c0的影响 - Python技术站
batch_first=True会对LSTM的输入输出的维度顺序有影响,但是对初始化h0和c0的维度顺序没有影响,也就是说,不管batch_first=True还是False,h0和c0的维度顺序都是:
关于LSTM的输入输出,可参考这篇博客。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:PyTorch LSTM,batch_first=True对初始化h0和c0的影响 - Python技术站