RuntimeError: expected scalar type Long but found Float 报错解决

问题描述

在运行别人代码时出现如下报错：

RuntimeError: expected scalar type Long but found Float
报错代码为：

data = list_text[1: 11]
        label = list_text[0]
        # print(data)
        # print(label)

        ##转tensor
        data = torch.tensor(data) #这行报错
        label = torch.tensor(label)
        print(data.shape)
        print(label.shape)

---

原因分析：

提示的错误是关于类型的问题，把列表变张量时，torch.long需要变torch.float类型

---

解决方案：

在如下代码位置加上，dtype显示地设置成torch.float

        data = torch.tensor(data, dtype=torch.float) #这行报错
        label = torch.tensor(label, dtype=torch.float)

报错解决！