list,numpy,tensor之间相互转换的方法

list,numpy,tensor之间相互转换的方法：

a=[[1,2],[3,4]]#list
print(a)
b=np.array(a)#list->numpy
print(b)
c=torch.tensor(a)#list->tensor
print(c)
print(b.tolist())#numpy->list
print(c.tolist())#tensor->list
print(torch.tensor(b))#numpy->tensor
print(torch.from_numpy(b))#同上
print(c.numpy())#tensor->numpy

对了，温馨提示，tensor可以在GPU上运行，其他两个都不可以，这就是为什么你用GPU运行的时候有时会报不是tensor的错误，必须先转化为tensor。
还有，GPU上的tensor不能直接转为numpy，需要先放到CPU上。

a=a.cpu()#放到CPU
a.numpy()#这才对。

放回GPU上？

a=a.cuda()#放到GPU

当然啦，cuda(),cpu()还有一种写法：

device=torch.device("cpu")#torch.device("cuda")
a=torch.tensor([1,2])
a.to(device)#放到CPU

list,numpy,tensor之间相互转换的方法相关推荐

c++ string 数组_PHP数组与字符串之间相互转换的方法
php中可以将数组转换成字符串的形式,也可以将字符串通过某个特定的规律转化成数组.那具体如何做呢?今天这篇文章就来说一说php中数组与字符串之间相互转换的方法. PHP 数组转换成字符串的方法 php ...
内核程序中进程的pid,handle,eprocess之间相互转换的方法
在内核程序开发中,我们常常需要取得某进程的pid或句柄,或者需要检索进程的eprocess结构,很多API函数需要的参数也不同,所以掌握pid<->handle<->eproc ...
C# DataTable 和List之间相互转换的方法
一. List<T>/IEnumerable转换到DataTable/DataView 方法一: /// <summary> /// Convert a List{T} to ...
学习Python必须要会的，在字符串，列表，元组三者之间相互转换的方法
1.字符串转换成列表 str='Lara1798' print(list(str))# 运行结果:['L', 'a', 'r', 'a', '1', '7', '9', '8'] 2.列表转换成字符串 ...
C++ 数字与字符串之间相互转换(多种方法)
使用系统提供的库函数 1.字符串传数字 (1).使用stoi() string s("12345"); long long a = stoi(s); cout << ...
PyTorch——torch.Tensor与np.ndarray（NumPy）之间的类型转换
1 前言今天在写 Digit Recognizer的代码~ 在对提交文件submission.csv进行写入操作的时候,总会有报错,其中一个原因是其实pandas对np.ndarray数据更加友好, ...
python image.open函数_PIL.Image.open和cv2.imread的比较与相互转换的方法
摘要 PIL.Image.open读入的是RGB顺序,而opencv中cv2.imread读入的是BGR通道顺序 .cv2.imread会显示图片更蓝一些. cv2.imread(path,读取方式) ...
使用C++实现YUV格式图像与RGB格式图像之间相互转换
使用C++实现YUV格式图像与RGB格式图像之间相互转换一.RGB与YUV转换公式 1.RGB转YUV 1)RGB转换亮度与色差信号公试: 2)归一化为YUV的转化公试为: 2.YUV转RGB 二. ...
.mat,.txt,.csv 数据转换为weka中的arff格式及matlab和Weka之间相互转换格式
在RUSBoost和SMOTEBoost中提供了csv转换为arff格式的方法,详见CSVtoARFF.m http://www.mathworks.com/matlabcentral/fileexc ...

list,numpy,tensor之间相互转换的方法

list,numpy,tensor之间相互转换的方法相关推荐

最新文章

热门文章