浅谈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的区别以及相互转换方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在windows系统上，重复性的操作可以用Python脚本来完成，其中常用的模块是win32gui、win32con、win32api，要使用这三个模块需要先安装pywin32。本文就为大家介绍了如何利用这些模块实现自动操作电脑，感兴趣的可以了解一下2022-11-11
今天小编就为大家分享一篇keras 特征图可视化实例(中间层)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了微软官方的Python插件，已经自带很多功能，下面是插件功能描述，其中部分内容我做了翻译,需要的朋友可以参考下2018-04-04
python Bamboolib库加速Pandas数据分析过程详解
这篇文章主要介绍了python Bamboolib库加速Pandas数据分析过程详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01
Python numpy.array()生成相同元素数组的示例
今天小编就为大家分享一篇Python numpy.array()生成相同元素数组的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
在进行网络爬虫时,经常会遇到网站的反爬机制,其中之一就是通过IP封禁来限制爬虫的访问,为了规避这种限制,使用动态IP代理是一种有效的方法,本文将介绍在Python爬虫中如何使用动态IP代理,以及一些防止被封的方法,文中有详细的代码讲解,需要的朋友可以参考下2023-11-11
这篇文章主要介绍了Python如何自定义邻接表图类问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
这篇文章主要介绍了Python Asyncio调度原理详情，Python.Asyncio是一个大而全的库，它包括很多功能，而跟核心调度相关的逻辑除了三种可等待对象外，还有其它一些功能，它们分别位于runners.py，base_event.py，event.py三个文件中2022-06-06
这篇文章主要介绍了基于Python第三方插件实现西游记章节标注汉语拼音的方法,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-05-05
下面小编就为大家分享一篇python初学之用户登录的实现过程(实例讲解)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2017-12-12
最新评论