pytorch tensor类型转换

迷南。 2022-11-19 09:47 262阅读 0赞

tensor转numpy

Tensor—->Numpy  使用 data.numpy()，data为Tensor变量  链接：[http://ddrv.cn/a/575619][http_ddrv.cn_a_575619]

![20210319180306872.png][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hcnRpbmtlaXRo_size_16_color_FFFFFF_t_70][]

另：

（1）数据类型转换

在Tensor后加 .long(), .int(), .float(), .double()等即可，也可以用.to()函数进行转换，所有的Tensor类型可参考[https://pytorch.org/docs/stable/tensors.html][https_pytorch.org_docs_stable_tensors.html]

（2）数据存储位置转换

CPU张量 —->  GPU张量，使用data.cuda()

GPU张量 —-> CPU张量，使用data.cpu()

（3）与numpy数据类型转换

Tensor—->Numpy  使用 data.numpy()，data为Tensor变量

Numpy —-> Tensor 使用 torch.from\_numpy(data)，data为numpy变量

（4）与Python数据类型转换

Tensor —-> 单个Python数据，使用data.item()，data为Tensor变量且只能为包含单个数据

Tensor —-> Python list，使用data.tolist()，data为Tensor变量，返回shape相同的可嵌套的

tensor中只有一个数时，直接转换成数 链接：[https://pytorch.org/docs/stable/tensors.html][https_pytorch.org_docs_stable_tensors.html]

>>> x = torch.tensor([[1]])
    >>> x
    tensor([[ 1]])
    >>> x.item()
    1
    >>> x = torch.tensor(2.5)
    >>> x
    tensor(2.5000)
    >>> x.item()
    2.5

[http_ddrv.cn_a_575619]: http://ddrv.cn/a/575619
[20210319180306872.png]: /images/20221119/de90ff7d2e2c4004b1f2507e312c4ceb.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hcnRpbmtlaXRo_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20210319180331424.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hcnRpbmtlaXRo,size_16,color_FFFFFF,t_70
[https_pytorch.org_docs_stable_tensors.html]: https://pytorch.org/docs/stable/tensors.html