PyTorch深度学习实战-2-pytorch和numpy互相转换

Posted esc_ai

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch深度学习实战-2-pytorch和numpy互相转换相关的知识,希望对你有一定的参考价值。

import torch
import numpy

# 创建tensor
torch_data = torch.Tensor([1,-2,3])

# Torch-->array
np_data = torch_data.numpy()

# array --> Torch

torch_data = torch.from_numpy(np_data)


以上是关于PyTorch深度学习实战-2-pytorch和numpy互相转换的主要内容,如果未能解决你的问题,请参考以下文章

PyTorch学习笔记:简介与基础知识

深度学习为什么选择Pytorch?史上最详细Pytorch入门教程

深度学习之30分钟快速入门PyTorch(附学习资源推荐)

嵌入式学深度学习:1Pytorch框架搭建

《动手学深度学习》环境配置(PyTorch版)

PyTorch学习笔记:PyTorch进阶训练技巧