PyTorch,LSTM,batch_first=True

torch.nn.lstm()接受的数据输入是(序列长度,batchsize,输入维数),使用batch_first=True,可以使lstm接受维度为(batchsize,序列长度,输入维数)的数据输入,同时,lstm的输出数据维度也会变为batchsize放在第一维(可参考这篇博客)。

 

posted @ 2020-08-13 12:03  Picassooo  阅读(2642)  评论(0)    收藏  举报