出现了TypeError: 'DataLoader' object is not subscriptable问题如何解决

Question

问题遇到的现象和发生背景

对LSTM的输入进行分组，之后想读取训练数据却报错

问题相关代码，请勿粘贴截图

seq_len = 3
batch_size = 2
data_transform = transforms.Lambda(lambda x: listToTensor(x))

dataset = DemoDatasetLSTM(data_, seq_len, transforms=data_transform)
data_loader = Data.DataLoader(dataset, batch_size, shuffle=False)

data = data_loader
data_x = data[:-1, :]  # 取除最后一行外的所有行
data_y = data[+1:, 0]  # 取第一列中除第一个元素以外的所有元素

运行结果及报错内容

Traceback (most recent call last):
File "G:/pythonProject2(LSTM input yanjiu)/input INS data.py", line 92, in
data_x = data[:-1, :] # 取除最后一行外的所有行
TypeError: 'DataLoader' object is not subscriptable

我的解答思路和尝试过的方法

我想要达到的结果

Answer 1

感觉是data_loader 没正确获取。
print(data_loader)
看看有啥输出