浅谈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小技巧之批量抓取美女图片
Jun 06 Python
Python程序设计入门(4)模块和包
Jun 16 Python
编写Python脚本批量下载DesktopNexus壁纸的教程
May 06 Python
剖析Python的Twisted框架的核心特性
May 25 Python
Python两个内置函数 locals 和globals(学习笔记)
Aug 28 Python
Python3用tkinter和PIL实现看图工具
Jun 21 Python
使用python实现抓取腾讯视频所有电影的爬虫
Apr 15 Python
利用anaconda保证64位和32位的python共存
Mar 09 Python
Django rest framework jwt的使用方法详解
Aug 08 Python
python字典和json.dumps()的遇到的坑分析
Mar 11 Python
python实现人机五子棋
Mar 25 Python
Python开发五子棋小游戏
May 02 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
虫族 Zerg 热键控制
2020/03/14 星际争霸
php5 图片验证码实现代码
2009/12/11 PHP
PHP时间戳与日期之间转换的实例介绍
2013/04/19 PHP
利用ajax和PHP实现简单的流程管理
2017/03/23 PHP
屏蔽F1~F12的快捷键的js函数
2010/05/06 Javascript
日期处理的js库(迷你版)--自建js库总结
2011/11/21 Javascript
js实现屏蔽默认快捷键调用自定义事件示例
2013/06/18 Javascript
js判断60秒以及倒计时示例代码
2014/01/24 Javascript
在Node.js应用中读写Redis数据库的简单方法
2015/06/30 Javascript
Vue2 监听属性改变watch的实例代码
2018/08/27 Javascript
Vue表单及表单绑定方法
2018/09/04 Javascript
vue+element模态框中新增模态框和删除功能
2019/06/11 Javascript
微信小程序加载机制及运行机制图解
2019/11/27 Javascript
对vue生命周期的深入理解
2020/12/03 Vue.js
[01:54]TI珍贵瞬间系列(五):压力
2020/08/29 DOTA
Python基于smtplib实现异步发送邮件服务
2015/05/28 Python
Python生成随机数组的方法小结
2017/04/15 Python
Python实现获取命令行输出结果的方法
2017/06/10 Python
Python模拟用户登录验证
2017/09/11 Python
python执行使用shell命令方法分享
2017/11/08 Python
Python 经典面试题 21 道【不可错过】
2018/09/21 Python
对python实现二维函数高次拟合的示例详解
2018/12/29 Python
python中取绝对值简单方法总结
2020/07/24 Python
Python实现七个基本算法的实例代码
2020/10/08 Python
基于python实现坦克大战游戏
2020/10/27 Python
html+css3实现的登录界面
2020/12/09 HTML / CSS
英国音乐设备和乐器商店:Gear4music
2017/10/16 全球购物
运动服饰每月订阅盒:Ellie
2018/04/29 全球购物
Abbott Lyon官网:女士手表、珠宝及配件
2020/12/26 全球购物
会计专业毕业生推荐信
2013/11/05 职场文书
春风行动实施方案
2014/03/28 职场文书
党的群众路线教育学习材料
2014/05/12 职场文书
实验室标语
2014/06/21 职场文书
教师党的群众路线教育实践活动剖析材料
2014/10/09 职场文书
2016年国庆节67周年活动总结
2016/04/01 职场文书
利用Redis实现点赞功能的示例代码
2022/06/28 Redis