代码报错解决

提示：RuntimeError: stack expects each tensor to be equal size, but got [12, 6000] at entry 0 and [10, 6000] at entry 9

在__getitem__方法中添加的代码用于处理情况，其中index + self.BATCH_SIZE 超出了数据集的长度。遇到的错误，"stack expects each tensor to be equal size"，很可能是因为当批量大小超过数据集末尾时，会出现不同大小的张量。你的批量大小为12，如果在数据集末尾剩下的样本少于12个，你会得到形状为 [12, 6000] 和 [n, 6000] 的张量，其中 'n' 小于12。

解决办法：在def __getitem__(self, index): 下添加以下代码

    def __getitem__(self, index): #dataloader给予一个索引给dataset，dataset返回一个元组x，y给dataloader
        num = len(self.x_data) # num = 792
        if index + self.BATCH_SIZE >= num:
            index = num - self.BATCH_SIZE   #如果 index + self.BATCH_SIZE 大于或等于 num，则 
        #通过 index = num - self.BATCH_SIZE - 1 将 index 调整为合适的值，以确保不超出数据集边 
        #界。即从倒数第12个开始索引
        x_batch = self.x_data[index:index + self.BATCH_SIZE]
        y_batch = self.y_data[index].long()