Tensor 和 NumPy 相互转换的实现

更新时间：2023年02月10日 10:10:52 作者：xzw96

本文主要介绍了Tensor 和 NumPy 相互转换的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

1. Tensor 转 NumPy

a = torch.ones(6)
b = a.numpy()
print(a, b)

a += 1
print(a, b)
b += 1
print(a, b)

tensor([1., 1., 1., 1., 1., 1.]) [1. 1. 1. 1. 1. 1.]
tensor([2., 2., 2., 2., 2., 2.]) [2. 2. 2. 2. 2. 2.]
tensor([3., 3., 3., 3., 3., 3.]) [3. 3. 3. 3. 3. 3.]

2. NumPy 数组转 Tensor

import numpy as np
a = np.ones(7)
b = torch.from_numpy(a)
print(a, b)

a += 1
print(a, b)
b += 1
print(a, b)

[1. 1. 1. 1. 1. 1. 1.] tensor([1., 1., 1., 1., 1., 1., 1.], dtype=torch.float64)
[2. 2. 2. 2. 2. 2. 2.] tensor([2., 2., 2., 2., 2., 2., 2.], dtype=torch.float64)
[3. 3. 3. 3. 3. 3. 3.] tensor([3., 3., 3., 3., 3., 3., 3.], dtype=torch.float64)

3. torch.tensor() 将 NumPy 数组转换成 Tensor

直接用torch.tensor()将NumPy数组转换成Tensor，该方法总是会进行数据拷贝，返回的Tensor和原来的数据不再共享内存。

import numpy as np
a = np.ones((2,3))
c = torch.tensor(a)
a += 1
print('a:',a)
print('c:',c)
print(id(a)==id(c))

a: [[2. 2. 2.]
 [2. 2. 2.]]
c: tensor([[1., 1., 1.],
        [1., 1., 1.]], dtype=torch.float64)
False

到此这篇关于Tensor 和 NumPy 相互转换的实现的文章就介绍到这了,更多相关Tensor 和 NumPy 相互转换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

python如何求数组连续最大和的示例代码
这篇文章主要介绍了python如何求数组连续最大和的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-02-02
Python pygame实现中国象棋单机版源码
今天给大家带来的是关于Python实战的相关知识,文章围绕着用Python pygame实现中国象棋单机版展开,文中有非常详细的代码示例,需要的朋友可以参考下
2021-06-06
Python实现批量提取PPT中的文字
这篇文章主要为大家详细介绍了如何使用Python中的pptx和docx库来将PPT中的文字提取到Word中,文中的示例代码讲解详细,有需要的可以参考下
2024-03-03
CentOS中升级Python版本的方法详解
本文给大家分享的是再centos系统中将Python版本从2.6升级到2.7的方法和升级过程中遇到的问题的处理，非常详细，有需要的小伙伴可以参考下
2017-07-07
在ubuntu16.04中将python3设置为默认的命令写法
这篇文章主要介绍了在ubuntu16.04中将python3设置为默认python的方法，非常不错，具有一定的参考借鉴价值，需要的朋友参考下吧
2018-10-10
python爬虫_自动获取seebug的poc实例
下面小编就为大家带来一篇python爬虫_自动获取seebug的poc实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-08-08
Python OpenCV 图像区域轮廓标记(框选各种小纸条)
这篇文章主要介绍了Python OpenCV 图像区域轮廓标记(框选各种小纸条)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-03-03
如何写好 Python 的 Lambda 函数
这篇文章主要介绍了如何写好 Python 的 Lambda 函数,Lambda 函数是 Python 中的匿名函数,下面文章通过介绍Lambda 函数的相关内容展开文章主题，需要的小伙伴可以参考一下
2022-03-03
Python入门教程(四)Python注释介绍
这篇文章主要介绍了Python入门教程(四)Python注释介绍,Python是一门非常强大好用的语言,也有着易上手的特性,本文为入门教程,需要的朋友可以参考下
2023-04-04
Python进阶篇之多线程爬取网页
这篇文章主要为大家介绍了Python进阶中利用多线程来爬取网页的示例实现及解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步
2021-10-10