pytorch如何增加维度_Pytorch中的LSTM:如何添加/更改序列长度维度?-问答-阿里云开发者社区-阿里云...

我在pytorch中运行LSTM，但据我所知，它只取序列长度= 1。当我将序列长度整形为4或其他数字时，就会得到输入和目标长度不匹配的错误。如果我同时对输入和目标进行整形，那么模型会抱怨它不接受多目标标签。 我的训练数据集有66512行和16839列，目标中有3个类别/类。我想使用批处理大小为200和序列长度为4，即在一个序列中使用4行数据。 请建议如何调整我的模型和/或数据，以便能够运行模型的各种序列长度(例如，4)。

batch_size=200

import torch

from torch.utils.data import TensorDataset

from torch.utils.data import DataLoader

train_target = torch.tensor(train_data[['Label1','Label2','Label3']].values.astype(np.float32))

train_target = np.argmax(train_target, axis=1)

train = torch.tensor(train_data.drop(['Label1','Label2','Label3'], axis = 1).values.astype(np.float32))

train_tensor = TensorDataset(train.unsqueeze(1), train_target)

train_loader = DataLoader(dataset = train_tensor, batch_size = batch_size, shuffle = True)

<