August 9, 2021

When we use RNN network (such as LSTM and GRU), we can use Embedding layer provided from PyTorch, and receive many different length sequence sentence input.

Problem

Today, when I was trying to use nn.Embedding() to splice and calculate sentence vectors by myself, I got an error like this on the DataLoader I prepared in advance: