在pytorch中使用GRU / LSTM从非顺序输入转换为顺序输出的正确方法

时间:2020-08-23 00:09:10

标签: pytorch lstm

我已经阅读了许多PyTorch教程,但是由于某些原因,我仍然从根本上迷上了一件事。

我的输入是一个400维张量,而我想要的输出是一个可变长度序列向量。

nn.LSTM和nn.GRU都接受并输出序列。那么如何以惯用的方式从400维张量(或大小(batch_size,400))变为大小(batch_size,max_seq_length,output_dim)呢?

0 个答案:

没有答案