pytorch 实现tensor与numpy数组转换

更新时间：2019年12月27日 15:52:30 作者：头发光了你就强了

今天小编就为大家分享一篇使用pytorch 实现tensor与numpy数组转换，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

看代码，tensor转numpy:

a = torch.ones(2,2)
b = a.numpy()
c=np.array(a) #也可以转numpy数组
print(type(a))
print(type(b))
print(a)
print(b)

输出为：

<class ‘torch.Tensor'>
<class ‘numpy.ndarray'>
tensor([[1., 1.],
[1., 1.]])
[[1. 1.]
[1. 1.]]

numpy转tensor：

import torch
import numpy as np

a = np.ones(5)
b = torch.from_numpy(a)
c=torch.Tensor(a) #也可以转pytorch Tensor
print(type(a))
print(type(b))
print(a)
print(b)

输出为：

<class ‘numpy.ndarray'>
<class ‘torch.Tensor'>
[1. 1. 1. 1. 1.]
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

可见pytorch的tensor对象与numpy数组是可以相互转换的，且numpy数组的默认类型是double

以上这篇pytorch 实现tensor与numpy数组转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python 列表,集合和字典的增删改查
这篇文章主要介绍了python 列表,集合和字典的增删改查,本文分别对他们一一说明,小编觉得这篇文章写的还不错，需要的朋友可以参考下，希望能够给你带来帮助
2021-10-10
Pytorch+PyG实现EdgeCNN过程示例详解
这篇文章主要为大家介绍了Pytorch+PyG实现EdgeCNN过程示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-04-04
在PyCharm下使用 ipython 交互式编程的方法
今天小编就为大家分享一篇在PyCharm下使用 ipython 交互式编程的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01
python微信公众号之关键词自动回复
这篇文章主要为大家详细介绍了python微信公众号之关键词自动回复，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-06-06
Python如何使用标准库tmpfile库创建临时文件
这篇文章主要介绍了Python如何使用标准库tmpfile库创建临时文件问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2024-02-02
Hadoop中的Python框架的使用指南
这篇文章主要介绍了Hadoop中的Python框架的使用指南,Hadoop一般使用复杂的Java操作,但通过该框架使得Python脚本操作Hadoop成为了可能,需要的朋友可以参考下
2015-04-04
使用tensorflow根据输入更改tensor shape
这篇文章主要介绍了使用tensorflow根据输入更改tensor shape，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
python的faker库用法
今天小编就为大家分享一篇python的faker库用法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-11-11
一篇文章带你了解python标准库--random模块
这篇文章主要给大家介绍了关于Python中random模块常用方法的使用教程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-08-08
Python爬虫UA伪装爬取的实例讲解
在本篇文章里小编给大家整理的是一篇关于Python爬虫UA伪装爬取的实例讲解内容，有需要的朋友们可以学习参考下。
2021-02-02