使用torch.tensor时报错

Question

使用torch.tensor时报错，代码如下：

def readCSV(filepath):
    file = open(filepath, encoding='utf-8-sig')
    context = file.read()
    list_result = context.split('\n')
    for i in range(len(list_result)):
        list_result[i] = list_result[i].split(',')
    return list_result
train = readCSV('train_data2.csv')
train_label = []
for line in train:
    train_label.append(line[0:1])

a = train_label[1:]
a = torch.tensor(a)

报错内容：

ValueError: too many dimensions 'str'

Answer 1

https://blog.csdn.net/qq_36653505/article/details/88416170