pytorch tensor操作：tensor与numpy转换

tensor转numpy

t = torch.ones(5)
print(f"t: {t}")
n = t.numpy()
print(f"n: {n}")

输出：

t: tensor([1., 1., 1., 1., 1.])
n: [1. 1. 1. 1. 1.]

cpu上的tensor可以和numpy array共享内存地址，改变其中的一个另一个也会改变

t.add_(1)
print(f"t: {t}")
print(f"n: {n}")

输出：

t: tensor([2., 2., 2., 2., 2.])
n: [2. 2. 2. 2. 2.]

可训练的tensor转numpy

t = torch.ones(5)
t_trained = t.clone().detach().requires_grad_(True)
print(f"t_trained: {t_trained}")
n = t_trained.detach().numpy()
print(f"n: {n}")

输出：

t_trained: tensor([1., 1., 1., 1., 1.], requires_grad=True)
n: [1. 1. 1. 1. 1.]

numpy转tensor

n = np.ones(5)
t = torch.from_numpy(n)
np.add(n, 1, out=n)
# n = np.add(n, 1) 不改变t，非in_place操作
print(f"t: {t}")
print(f"n: {n}")

输出：

t: tensor([2., 2., 2., 2., 2.], dtype=torch.float64)
n: [2. 2. 2. 2. 2.]

pytorch tensor操作：tensor与numpy转换相关推荐

PyTorch 安装和基本运算— Tensor 的数据类型（浮点型、整型、随机浮点型等）、基本运算（绝对值、求和、裁剪、求商、求积、求幂等）、Tensor 与 Numpy 转换
PyTorch 的安装可以到官网 https://pytorch.org/,选择适合自己机器以及安装方式,执行对应的命令即可. 除了安装 PyTorch 之外,建议也安装 torchvision 包. ...
pytorch:tensor与numpy转换 .cpu.numpy()和.numpy() torch.from_numpy VS torch.Tensor
1.tensor to numpy 1.1代码 .numpy() import torch a0 = torch.ones(6) print(a0) 运行结果: tensor([1., 1., 1., ...
Pytorch List Tensor转Tensor,,reshape拼接等操作
Pytorch List Tensor转Tensor,reshape拼接等操作持续更新一些常用的Tensor操作,比如List,Numpy,Tensor之间的转换,Tensor的拼接,维度的变换等操 ...
pytorch 图像与tensor转换
Pytorch中Tensor与各种图像格式的相互转化在pytorch中经常会遇到图像格式的转化,例如将PIL库读取出来的图片转化为Tensor,亦或者将Tensor转化为numpy格式的图片.而且使 ...
pytorch基础操作学习笔记（autograd，Tensor）
简述简单讲讲关于torch.autograd内容(因为我也有点菜) 文章目录简述简单讲讲 Tensor 介绍Tensor 创建Tensor 获取Tensor数据规模将tensor转成其他数据类 ...
pytorch新手需要注意的隐晦操作Tensor,max,gather
文章移到:https://oldpan.me/archives/pytorch-rookie-care-tensor-max-gather pytorch中有很多操作比较隐晦,需要仔细研究结合一些例子 ...
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换 1, 创建pytorch 的Tensor张量: torch.rand((3,224,224)) #创建随机值的三维张量,大小为 ...
Python数据类型、Numpy数据类型和Pytorch中的tensor类型间的相互转化
数据类型包括Python数据类型.Numpy数据类型和Pytorch中的tensor,Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型. 一.Python数据类型 Pytho ...
Pytorch | 报错The given NumPy array is not writeable,and PyTorch does not support non-writeable tensor
Pytorch | 报错 The given NumPy array is not writeable,and PyTorch does not support non-writeable tenso ...
PyTorch: tensor操作(一) contiguous
本文目录 tensor在内存中的存储信息区和存储区 shape && stride contiguous 什么时候用contiguous呢? 为什么要用contiguous 为什么c ...

pytorch tensor操作：tensor与numpy转换

tensor转numpy

可训练的tensor转numpy

numpy转tensor

pytorch tensor操作：tensor与numpy转换相关推荐

最新文章

热门文章