Python列表与张量之间的转换

 在学习深度学习时经常遇到列表数据与张量数据之间的转换问题，记录下来也分享出来供大家参考:

import numpy as np
import torch

#列表转张量：
def ToTensor(data):
    data_n = np.array(data)
    data_out = torch.tensor(data_n, dtype=torch.float).reshape(len(data_n), 1, 512)
    return data_out
#张量转列表：
def ToList(data):
    data_squeeze = data.squeeze()
    data_list = data_squeeze[:].detach().numpy().tolist()
    return data_list