将python列表的python列表转换为pytorch张量

Question

将列表的python列表转换为PyTorch张量的常规方法是什么？

a = [0,0]
b = [1,1]
c = [2]
c = [a, b, c]

我希望将c转换为扁平的Torch张量，如下所示：

torch([ 0, 0, 1, 1, 2])

Answer 1

您可以先使用Python拼合列表：

flat_list = [item for sublist in c for item in sublist]

并创建您的张量：

flattened_tensor = torch.FloatTensor(flat_list)