浅谈pytorch和Numpy的区别以及相互转换方法

更新时间：2018年07月26日 14:33:33 作者：qq_34535410

今天小编就为大家分享一篇浅谈pytorch和Numpy的区别以及相互转换方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如下所示：

# -*- coding: utf-8 -*-
# @Time  : 2018/1/17 16:37
# @Author : Zhiwei Zhong
# @Site  : 
# @File  : Numpy_Pytorch.py
# @Software: PyCharm

import torch
import numpy as np

np_data = np.arange(6).reshape((2, 3))

# numpy 转为 pytorch格式

torch_data = torch.from_numpy(np_data)
print(
  '\n numpy', np_data,
  '\n torch', torch_data,
)
'''
 numpy [[0 1 2]
 [3 4 5]] 
 torch 
 0 1 2
 3 4 5
[torch.LongTensor of size 2x3]
'''
# torch 转为numpy
tensor2array = torch_data.numpy()
print(tensor2array)
"""
[[0 1 2]
 [3 4 5]]
"""
# 运算符
# abs 、 add 、和numpy类似
data = [[1, 2], [3, 4]]
tensor = torch.FloatTensor(data)    # 转为32位浮点数，torch接受的都是Tensor的形式，所以运算前先转化为Tensor
print(
  '\n numpy', np.matmul(data, data),
  '\n torch', torch.mm(tensor, tensor)    # torch.dot()是点乘
)
'''
 numpy [[ 7 10]
 [15 22]] 
 torch 
 7 10
 15 22
[torch.FloatTensor of size 2x2]
'''

以上这篇浅谈pytorch和Numpy的区别以及相互转换方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python列表元素常见操作简单示例
这篇文章主要介绍了Python列表元素常见操作,结合简单示例形式分析了Python针对列表元素的打印、添加、删除、修改、排序等相关操作技巧与注意事项,需要的朋友可以参考下
2019-10-10
Python3.5字符串常用操作实例详解
这篇文章主要介绍了Python3.5字符串常用操作,结合实例形式总结分析了Python3.5字符串输入、输出、格式化、切片以及各种常用操作函数相关使用技巧,需要的朋友可以参考下
2019-05-05
我用Python抓取了7000 多本电子书案例详解
这篇文章主要介绍了我用Python抓取了7000 多本电子书案例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-03-03
Python使用Beautiful Soup实现解析网页
在这篇文章中，我们将介绍如何使用 Python 编写一个简单的网络爬虫，以获取并解析网页内容。我们将使用 Beautiful Soup 库，它是一个非常强大的库，用于解析和操作 HTML 和 XML 文档。让我们开始吧
2023-05-05
numpy.unique()使用方法
本文主要介绍了numpy.unique()使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-02-02
Python配置同花顺全数据接口教程详解
这篇文章主要为大家详细介绍了Python配置同花顺全数据接口的教程，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2022-01-01
Python实现的拟合二元一次函数功能示例【基于scipy模块】
这篇文章主要介绍了Python实现的拟合二元一次函数功能,结合实例形式分析了Python基于scipy模块进行二元一次函数拟合相关科学运算操作实现技巧,需要的朋友可以参考下
2018-05-05
关于vscode 默认添加python项目的源目录路径到执行环境的问题
这篇文章主要介绍了vscode 默认添加python项目的源目录路径到执行环境,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-02-02
在Python的Bottle框架中使用微信API的示例
这篇文章主要介绍了在Python的Bottle框架中使用微信API的示例,作者还在文中给出了一个生成的微信可扫描的二维码图,需要的朋友可以参考下
2015-04-04
python批量将PDF文件转换成图片的实现代码
这篇文章使用python编写了一个小脚本,目的是为了实现批量将PDF文件转换成图片,文中有详细的实现代码,对我们的学习或工作有一定的帮助,感兴趣的小伙伴可以参考阅读一下
2023-08-08