将python列表转换为PyTorch张量的传统方法是什么?
a = [0,0] b = [1,1] c = [2] c = [a, b, c]
我想把c转换成一个扁平的火炬张量,如下所示:
torch([ 0, 0, 1, 1, 2])
您可以先在Python中展开列表:
flat_list = [item for sublist in c for item in sublist]
flattened_tensor = torch.FloatTensor(flat_list)
