标签: pytorch lstm
我已经阅读了许多PyTorch教程,但是由于某些原因,我仍然从根本上迷上了一件事。
我的输入是一个400维张量,而我想要的输出是一个可变长度序列向量。
nn.LSTM和nn.GRU都接受并输出序列。那么如何以惯用的方式从400维张量(或大小(batch_size,400))变为大小(batch_size,max_seq_length,output_dim)呢?