浅谈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获取本机外网ip的方法
Apr 15 Python
使用Django的模版来配合字符串翻译工作
Jul 27 Python
在Python的Flask框架中构建Web表单的教程
Jun 04 Python
python实现xlsx文件分析详解
Jan 02 Python
对numpy.append()里的axis的用法详解
Jun 28 Python
Python 经典面试题 21 道【不可错过】
Sep 21 Python
Python爬虫文件下载图文教程
Dec 23 Python
手机使用python操作图片文件(pydroid3)过程详解
Sep 25 Python
python实现文件批量编码转换及注意事项
Oct 14 Python
Python 实现加密过的PDF文件转WORD格式
Feb 04 Python
django在保存图像的同时压缩图像示例代码详解
Feb 11 Python
python批量生成身份证号到Excel的两种方法实例
Jan 14 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 多关键字 高亮显示实现代码
2012/04/23 PHP
php无限极分类实现的两种解决方法
2013/04/28 PHP
php的ddos攻击解决方法
2015/01/08 PHP
抛弃 PHP 代价太高
2016/04/26 PHP
什么是PHP7中的孤儿进程与僵尸进程
2019/04/14 PHP
php文件后缀不强制为.php的实操方法
2019/09/18 PHP
javascript getElementsByTagName
2011/01/31 Javascript
动态加载js、css等文件跨iframe实现
2014/02/24 Javascript
JS生成不重复随机数组的函数代码
2014/06/10 Javascript
javascript实现根据时间段显示问候语的方法
2015/06/18 Javascript
JavaScript+CSS实现的可折叠二级菜单实例
2016/02/29 Javascript
JQuery validate插件验证用户注册信息
2016/05/11 Javascript
[原创]jQuery常用的4种加载方式分析
2016/07/25 Javascript
详解js中call与apply关键字的作用
2016/11/21 Javascript
详解angular2封装material2对话框组件
2017/03/03 Javascript
快速将Vue项目升级到webpack3的方法步骤
2017/09/14 Javascript
基于Vue的延迟加载插件vue-view-lazy
2018/05/21 Javascript
小程序ios音频播放没声音问题的解决
2018/07/11 Javascript
基于javascript实现碰撞检测
2020/03/12 Javascript
[01:38]女王驾到——至宝魔廷新尊技能&特效展示
2020/06/16 DOTA
Python中函数的参数传递与可变长参数介绍
2015/06/30 Python
分析python切片原理和方法
2017/12/19 Python
python3处理word文档实例分析
2020/12/01 Python
python drf各类组件的用法和作用
2021/01/12 Python
介绍一下Linux文件的记录形式
2012/04/18 面试题
日语翻译个人求职的自我评价
2013/10/14 职场文书
精彩的推荐信范文
2013/11/26 职场文书
羽毛球比赛策划方案
2014/06/13 职场文书
化工专业自荐书
2014/06/16 职场文书
2014年社区党建工作汇报材料
2014/11/02 职场文书
2014年驻村干部工作总结
2014/11/17 职场文书
2015年消费者权益日活动总结
2015/02/09 职场文书
音乐教师求职信范文
2015/03/20 职场文书
自荐信模板大全
2015/03/27 职场文书
2015初一年级组工作总结
2015/07/24 职场文书
毕业班工作总结
2015/08/10 职场文书