百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术分类 > 正文

一文读懂pytorch基础语法

ztj100 2025-03-11 23:54 8 浏览 0 评论

一.定义张量tensor 张量

一种数据类型,可以是一到多维矩阵,在机器学习中作为主要输入输出数据类型使用

1.自定义数值的张量torch.tensor([[2.1,12.2,2.121],[2.1,12.2,2.121]],dtype=float)

2.张量ndim,shape ,size属性,ndim表示张量的维度,shape=size表示张量的形状

3.定义全1张量test2 = torch.ones(8,8)

4.定义全零张量test3 = torch.zeros(2,9)

5.定义随机张量test4 = torch.rand(1,20)

6.定义随机整型张量test5 = torch.randint(0,10,(2,3))

7.定义正态分布张量test6 = torch.randn(1,9)

二.张量的基本运算

1.改变张量(矩阵)的维度,修改前后矩阵元素总数不变test6 = test6.view(3,3)

2.获取tensor第n个元素

test6[1][1]:获取二维矩阵第1行第一列元素

test6[1] 若是一维矩阵则使用获取第一个元素,

test[1:10]获取从索引1到10的所有元素

test[:10]获取10以前的所有数据

test[10:]获取索引10以后的所有数据 tensor_b[[1,2],[1,1]] 根据索引获取数据 3.test6[1].item()获取指定索引元素的值

4.pytorch张量和numpy数组之前的转化

numarr = numpy.array(test6)

tensor转化为numpy 数组 tensor = torch.tensor(nparr) numpy数组转化为tensor张量

5.基本运算

加法运算: tensor_c = tensor_a+tensor_b ,将矩阵对应元素分别相加, 加法运算的另外两种不常用写法 res = torch.zeros(3,3,dtype=int) torch.add(tensor_a,tensor_b,out=res) 计算结果存储在res变量中 tensor_c = torch.add(tensor_a,tensor_b)

减法运算: tensor_c = tensor_a+tensor_b ,将矩阵对应元素分别相减

除法运算:tensor_c = tensor_a/tensor_b,将矩阵对应元素分别相除

取余运算:tensor_c = tensor_a%tensor_b,对应元素相除取余数

乘法运算:tensor_c = tensor_a*tensor_b,非矩阵乘法,对应元素相乘

矩阵相乘:使用matmul函数tensor_c = torch.matmul(tensor_a,tensor_b)

对矩阵求和:tensor = torch.sum(tensor_c) 所有元素相加

求最大值最小值等:tensor = torch.max/min(tensor_c)

求最大值最小值所在的索引 index= torch.argmax(tensor_c) arg+max/min

求均值:tensor = torch.mean(tensor_b) tensor_b必须为float类型 获得矩阵中位数:tensor = torch.median(tensor_b) 开方:tensor = torch.sqrt(tensor_b)

幂运算:tensor = torch**3

三.自动求导

x = torch.tensor([6,3],dtype=float,requires_grad=True) #用于说明当前量是否需要在计算中保留对应的梯度信息

y=x**3+1

y.backward(torch.ones_like(x)) #若输出是标量则y.backward()

x.grad 注:y=x^3+1 求导:根据求导公式得出yd= x^3`+1` =>yd = 3x^2

文章来源:
https://www.ixiera.com/web/kehu/zsk/427.html

了解更多

相关推荐

Whoosh,纯python编写轻量级搜索工具

引言在许多应用程序中,搜索功能是至关重要的。Whoosh是一个纯Python编写的轻量级搜索引擎库,可以帮助我们快速构建搜索功能。无论是在网站、博客还是本地应用程序中,Whoosh都能提供高效的全文搜...

如何用Python实现二分搜索算法(python二分法查找代码)

如何用Python实现二分搜索算法二分搜索(BinarySearch)是一种高效的查找算法,适用于在有序数组中快速定位目标值。其核心思想是通过不断缩小搜索范围,每次将问题规模减半,时间复杂度为(O...

路径扫描 -- dirsearch(路径查找器怎么使用)

外表干净是尊重别人,内心干净是尊重自己,干净,在今天这个时代,应该是一种极高的赞美和珍贵。。。----网易云热评一、软件介绍Dirsearch是一种命令行工具,可以强制获取web服务器中的目录和文件...

78行Python代码帮你复现微信撤回消息!

来源:悟空智能科技本文约700字,建议阅读5分钟。本文基于python的微信开源库itchat,教你如何收集私聊撤回的信息。...

从零开始学习 Python!2《进阶知识》 Python进阶之路

欢迎来到Python学习的进阶篇章!如果你说已经掌握了基础语法,那么这篇就是你开启高手之路的大门。我们将一起探讨面向对象编程...

白帽黑客如何通过dirsearch脚本工具扫描和收集网站敏感文件

一、背景介绍...

Python之txt数据预定替换word预定义定位标记生成word报告(四)

续接Python之txt数据预定替换word预定义定位标记生成word报告(一)https://mp.toutiao.com/profile_v4/graphic/preview?pgc_id=748...

假期苦短,我用Python!这有个自动回复拜年信息的小程序

...

Python——字符串和正则表达式中的反斜杠('\')问题详解

在本篇文章里小编给大家整理的是关于Python字符串和正则表达式中的反斜杠('\')问题以及相关知识点,有需要的朋友们可以学习下。在Python普通字符串中在Python中,我们用'\'来转义某些普通...

Python re模块:正则表达式综合指南

Python...

Python中re模块详解(rem python)

在《...

python之re模块(python re模块sub)

re模块一.re模块的介绍1.什么是正则表达式"定义:正则表达式是一种对字符和特殊字符操作的一种逻辑公式,从特定的字符中,用正则表达字符来过滤的逻辑。(也是一种文本模式;)2、正则表达式可以帮助我们...

MySQL、PostgreSQL、SQL Server 数据库导入导出实操全解

在数字化时代,数据是关键资产,数据库的导入导出操作则是连接数据与应用场景的桥梁。以下是常见数据库导入导出的实用方法及代码,包含更多细节和特殊情况处理,助你应对各种实际场景。一、MySQL数据库...

Zabbix监控系统系列之六:监控 mysql

zabbix监控mysql1、监控规划在创建监控项之前要尽量考虑清楚要监控什么,怎么监控,监控数据如何存储,监控数据如何展现,如何处理报警等。要进行监控的系统规划需要对Zabbix很了解,这里只是...

mysql系列之一文详解Navicat工具的使用(二)

本章内容是系列内容的第二部分,主要介绍Navicat工具的使用。若查看第一部分请见:...

取消回复欢迎 发表评论: