相信这是每个做deep learning的朋友都有可能遇到的问题。例如,我们有一些其他方法生成的embedding,那么通常是存在文本文件中的,需要进行处理并输出到文本文件,那这中间就牵扯到这些对应的转换,用一个例子具体总结一下:
import torch
import numpy as np
string_list=['1.0','2.0','3.0']
float_list=list(map(float,string_list))
tensor=torch.Tensor(float_list)
ndarray=tensor.cpu().numpy()
ndarray=np.array(float_list)
new_list=ndarray.tolist()
tensor=torch.from_numpy(ndarray).to(device).float()
print(tensor)
参考了下列朋友的总结:
https://www.cnblogs.com/jiading/p/12191004.html
https://blog.csdn.net/weixin_44489823/article/details/108212776