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


Posted in Python onJuly 26, 2018

如下所示:

# -*- 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中unittest实现简单的单元测试实例详解
Jan 09 Python
Python命令启动Web服务器实例详解
Feb 23 Python
Python 登录网站详解及实例
Apr 11 Python
Python 网页解析HTMLParse的实例详解
Aug 10 Python
利用pandas读取中文数据集的方法
Jul 25 Python
Django 拆分model和view的实现方法
Aug 16 Python
python代码实现TSNE降维数据可视化教程
Feb 28 Python
keras load model时出现Missing Layer错误的解决方式
Jun 11 Python
vscode调试django项目的方法
Aug 06 Python
Python实现Canny及Hough算法代码实例解析
Aug 06 Python
Python如何telnet到网络设备
Feb 18 Python
Python绘制分类图的方法
Apr 20 Python
pytorch: tensor类型的构建与相互转换实例
Jul 26 #Python
pytorch中tensor的合并与截取方法
Jul 26 #Python
Python爬虫框架Scrapy常用命令总结
Jul 26 #Python
Python退火算法在高次方程的应用
Jul 26 #Python
Python爬虫框架Scrapy基本用法入门教程
Jul 26 #Python
Tensorflow 合并通道及加载子模型的方法
Jul 26 #Python
解决tensorflow测试模型时NotFoundError错误的问题
Jul 26 #Python
You might like
php实现微信公众号无限群发
2015/10/11 PHP
Laravel利用gulp如何构建前端资源详解
2018/06/03 PHP
浅谈tudou土豆网首页图片延迟加载的效果
2010/06/23 Javascript
解决火狐浏览器下JS setTimeout函数不兼容失效不执行的方法
2012/11/14 Javascript
js监听鼠标点击和键盘点击事件并自动跳转页面
2014/09/24 Javascript
jquery+正则实现统一的表单验证
2015/09/20 Javascript
jquery显示隐藏元素的实现代码
2016/05/19 Javascript
jquery删除table当前行的实例代码
2016/10/07 Javascript
JS实现简单的天数计算器完整实例
2017/04/28 Javascript
Vue.js实现一个todo-list的上移下移删除功能
2017/06/26 Javascript
AngularJS实现select的ng-options功能示例
2017/07/12 Javascript
vue keep-alive 动态删除组件缓存的例子
2019/11/04 Javascript
原生js实现点击轮播切换图片
2020/02/11 Javascript
[10:21]2018DOTA2国际邀请赛寻真——Winstrike
2018/08/11 DOTA
Django权限机制实现代码详解
2018/02/05 Python
一个简单的python爬虫程序 爬取豆瓣热度Top100以内的电影信息
2018/04/17 Python
解决nohup重定向python输出到文件不成功的问题
2018/05/11 Python
python基础学习之如何对元组各个元素进行命名详解
2018/07/12 Python
python+mysql实现个人论文管理系统
2019/10/25 Python
Anaconda 查看、创建、管理和使用python环境的方法
2019/12/03 Python
python3 动态模块导入与全局变量使用实例
2019/12/22 Python
tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this T
2020/06/22 Python
python/golang实现循环链表的示例代码
2020/09/14 Python
python数据抓取3种方法总结
2021/02/07 Python
HTML5不支持frameset的两种解决方法
2016/11/14 HTML / CSS
Burberry英国官网:英国标志性奢侈品牌
2017/03/29 全球购物
荷兰牛仔裤网上商店:Jeans Centre
2018/04/03 全球购物
蒙蒂塞罗商店:Monticello Shop
2018/11/25 全球购物
毕业生的求职信范文分享
2013/12/04 职场文书
超市优秀员工事迹材料
2014/05/01 职场文书
活动总结模板
2014/05/09 职场文书
房地产广告策划方案
2014/05/15 职场文书
2014年办公室文秘工作总结
2014/12/09 职场文书
优秀教师先进事迹材料
2014/12/15 职场文书
刑事撤诉申请书
2015/05/18 职场文书
IDEA中sout快捷键无效问题的解决方法
2022/07/23 Java/Android