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