浅谈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复制文件的方法实例详解
May 22 Python
用python写一个windows下的定时关机脚本(推荐)
Mar 21 Python
对python 判断数字是否小于0的方法详解
Jan 26 Python
Python OpenCV 调用摄像头并截图保存功能的实现代码
Jul 02 Python
Python基于OpenCV实现人脸检测并保存
Jul 23 Python
django rest framework 实现用户登录认证详解
Jul 29 Python
python实现文件的分割与合并
Aug 29 Python
详解python statistics模块及函数用法
Oct 27 Python
简单了解pytest测试框架setup和tearDown
Apr 14 Python
Pycharm创建python文件自动添加日期作者等信息(步骤详解)
Feb 03 Python
python实现调用摄像头并拍照发邮箱
Apr 27 Python
Python使用plt.boxplot()函数绘制箱图、常用方法以及含义详解
Aug 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中文本数据翻页(留言本翻页)
2006/10/09 PHP
PHP4 与 MySQL 交互使用
2006/10/09 PHP
一个从别的网站抓取信息的例子(域名查询)
2006/10/09 PHP
PHP游戏编程25个脚本代码
2011/02/08 PHP
PHP数组游标实现对数组的各种操作详解
2016/01/26 PHP
PHP chunk_split()函数讲解
2019/02/12 PHP
jQuery checkbox全选/取消全选实现代码
2009/11/14 Javascript
javascript写的日历类(基于pj)
2010/12/28 Javascript
jQuery实现移动 和 渐变特效的点击事件
2015/02/26 Javascript
JavaScript优化专题之Loading and Execution加载和运行
2016/01/20 Javascript
jquery dialog获取焦点的方法
2017/02/09 Javascript
如何解决vue与传统jquery插件冲突
2017/03/20 Javascript
JS实现禁止用户使用Ctrl+鼠标滚轮缩放网页的方法
2017/04/28 Javascript
JavaScript实现移动端轮播效果
2017/06/06 Javascript
详解在vue-cli项目中使用mockjs(请求数据删除数据)
2017/10/23 Javascript
Vue封装Swiper实现图片轮播效果
2018/02/06 Javascript
NodeJS使用Range请求实现下载功能的方法示例
2018/10/12 NodeJs
使用vue-router切换页面时,获取上一页url以及当前页面url的方法
2019/05/06 Javascript
Webpack中SplitChunksPlugin 配置参数详解
2020/03/24 Javascript
基于vue中的scoped坑点解说
2020/09/04 Javascript
微信小程序实现电影App导航和轮播
2020/11/30 Javascript
CentOS 8.2服务器上安装最新版Node.js的方法
2020/12/16 Javascript
Python深入06——python的内存管理详解
2016/12/07 Python
pyqt5移动鼠标显示坐标的方法
2019/06/21 Python
python3使用腾讯企业邮箱发送邮件的实例
2019/06/28 Python
Python @property使用方法解析
2019/09/17 Python
python爬虫爬取网页数据并解析数据
2020/09/18 Python
Python常用断言函数实例汇总
2020/11/30 Python
python反编译教程之2048小游戏实例
2021/03/03 Python
高级Java程序员面试要点
2013/08/02 面试题
精选干货:Java精选笔试题附答案
2014/01/18 面试题
个人求职简历的自我评价范文
2013/10/09 职场文书
给市场的环保建议书
2014/05/14 职场文书
大学英语专业求职信
2014/06/21 职场文书
幼儿园2014年度工作总结
2014/11/10 职场文书
react中props 的使用及进行限制的方法
2021/04/28 Javascript