Tensor 和 NumPy 相互转换

我们很容易用numpy()和from_numpy()将Tensor和NumPy中的数组相互转换。但是需要注意的一点是：这两个函数所产生的Tensor和NumPy中的数组共享相同的内存（所以他们之间的转换很快），改变其中一个时另一个也会改变！

文章目录

Tensor 和 NumPy 相互转换
- 1. Tensor 转 NumPy
- 2. NumPy 数组转 Tensor
- 3. torch.tensor() 将 NumPy 数组转换成 Tensor

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 相互转换相关推荐

PyTorch 笔记（10）— Tensor 与 NumPy 相互转换、两种共享内存以及两者的广播法则
Tensor 与 NumPy 有很高的相似性,彼此之间的互操作也非常简单有效,需要注意的是 Tensor 与 NumPy 共享内存,由于 NumPy 历史悠久,所以遇到 Tensor 不支持的操作时, ...
Variable、Tensor、Numpy的转换
Tensor <--> Numpy Tensor与Numpy之间可以相互转换: 代码: [plain] view plaincopy import torch import numpy a ...
python代码转换为pytorch_python基础教程Pytorch之Tensor和Numpy之间的转换的实现方法...
为什么要相互转换: 1. 要对tensor进行操作,需要先启动一个Session,否则,我们无法对一个tensor比如一个tensor常量重新赋值或是做一些判断操作,所以如果将它转化为numpy数组就 ...
PyTorch 安装和基本运算— Tensor 的数据类型（浮点型、整型、随机浮点型等）、基本运算（绝对值、求和、裁剪、求商、求积、求幂等）、Tensor 与 Numpy 转换
PyTorch 的安装可以到官网 https://pytorch.org/,选择适合自己机器以及安装方式,执行对应的命令即可. 除了安装 PyTorch 之外,建议也安装 torchvision 包. ...
2021-07-26 TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the
can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. 意思是:如果 ...
can‘t convert cuda:0 device type tensor to numpy
can't convert cuda:0 device type tensor to numpy x是list,list内容是tensor cuda 代码: x = np.array(x) 则会报错: ...
Pytorch中的variable, tensor与numpy相互转化
来源:https://blog.csdn.net/m0_37592397/article/details/88327248 1.将numpy矩阵转换为Tensor张量 sub_ts = torch.f ...
pytorch Tensor转numpy并解决RuntimeError: Can‘t call numpy() on Tensor that requires grad.报错
解决方法转numpy时使用Tensor.detach().numpy(): a = torch.ones(5) b = a.detach().numpy() print(b) 问题解析当计算中的t ...
Pytorch:Tensor和Numpy
3.1.2 Tensor和Numpy Tensor和Numpy数组之间具有很高的相似性,彼此之间的互操作也非常简单高效.需要注意的是,Numpy和Tensor共享内存.由于Numpy历史悠久,支持丰富 ...

Tensor 和 NumPy 相互转换

Tensor 和 NumPy 相互转换

文章目录

1. Tensor 转 NumPy

2. NumPy 数组转 Tensor

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

Tensor 和 NumPy 相互转换相关推荐

最新文章

热门文章