В моей модели нейронной сети я представляю предложение из 8 слов с 8x256
размерной матрицей вложения.Я хочу передать его в LSTM в качестве входных данных, где LSTM в качестве входных данных принимает вложение одного слова за раз и обрабатывает его.В соответствии с документацией pytorch , ввод должен быть в форме (seq_len, batch, input_size)
.Как правильно преобразовать мои данные в нужную форму?Я не хочу путать числа по ошибке.Я новичок в вычислениях PyTorch и мажорных строках, поэтому я хотел спросить об этом здесь.Я делаю это следующим образом, это правильно?
x = torch.rand(8,256)
lstm_input = torch.reshape(x,(8,1,256))