浅谈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网络编程学习笔记(九):数据库客户端 DB-API
Jun 09 Python
python实现简单tftp(基于udp协议)
Jul 30 Python
django框架之cookie/session的使用示例(小结)
Oct 15 Python
python腾讯语音合成实现过程解析
Aug 01 Python
Python稀疏矩阵及参数保存代码实现
Apr 18 Python
使用pyplot.matshow()函数添加绘图标题
Jun 16 Python
python爬虫判断招聘信息是否存在的实例代码
Nov 20 Python
用python爬虫批量下载pdf的实现
Dec 01 Python
Python数据分析库pandas高级接口dt的使用详解
Dec 11 Python
用python实现监控视频人数统计
May 21 Python
python如何利用cv2模块读取显示保存图片
Jun 04 Python
总结Python使用过程中的bug
Jun 18 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
Terran热键控制
2020/03/14 星际争霸
php基本函数汇总
2015/07/09 PHP
php+ajax简单实现全选删除的方法
2016/12/06 PHP
jquery.validate使用攻略 第一部
2010/07/01 Javascript
jquery提升性能最佳实践小结
2010/12/06 Javascript
DIV+CSS+JS不间断横向滚动实现代码
2013/03/19 Javascript
JQuery中对Select的option项的添加、删除、取值
2013/08/25 Javascript
js实现刷新iframe的方法汇总
2015/04/27 Javascript
js实现两点之间画线的方法
2015/05/12 Javascript
浅析Javascript中bind()方法的使用与实现
2016/04/29 Javascript
JS判断日期格式是否合法的简单实例
2016/07/11 Javascript
php register_shutdown_function函数详解
2017/07/23 Javascript
JavaScript EventEmitter 背后的秘密 完整版
2018/03/29 Javascript
利用nodeJs anywhere搭建本地服务器环境的方法
2018/05/12 NodeJs
Angular4 反向代理Details实践
2018/05/30 Javascript
jsonp实现百度下拉框功能的方法分析
2019/05/10 Javascript
vue源码中的检测方法的实现
2019/09/26 Javascript
ElementUI Tree 树形控件的使用并给节点添加图标
2020/02/27 Javascript
Vue的v-model的几种修饰符.lazy,.number和.trim的用法说明
2020/08/05 Javascript
[01:06:54]DOTA2-DPC中国联赛 正赛 SAG vs DLG BO3 第二场 2月28日
2021/03/11 DOTA
python中安装Scrapy模块依赖包汇总
2017/07/02 Python
python爬取淘宝商品详情页数据
2018/02/23 Python
python中将zip压缩包转为gz.tar的方法
2018/10/18 Python
Django异步任务之Celery的基本使用
2019/03/23 Python
python中的单引号双引号区别知识点总结
2019/06/23 Python
PyQt5实现暗黑风格的计时器
2019/07/29 Python
tensorflow实现测试时读取任意指定的check point的网络参数
2020/01/21 Python
Pytorch .pth权重文件的使用解析
2020/02/14 Python
用HTML5制作烟火效果的教程
2015/05/12 HTML / CSS
国际象棋商店:The Chess Store
2018/07/09 全球购物
美国滑雪板和装备购物网站:Skis.com
2018/12/20 全球购物
美国婴儿用品及配件购买网站:Munchkin
2019/04/03 全球购物
幼儿园评语大全
2014/04/17 职场文书
高校优秀辅导员事迹材料
2014/05/07 职场文书
乡镇法制宣传日活动总结
2015/05/05 职场文书
2016学校元旦晚会经典开场白台词
2015/12/03 职场文书