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

一文读懂pytorch基础语法

ztj100 2025-03-11 23:54 11 浏览 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

了解更多

相关推荐

使用 Pinia ORM 管理 Vue 中的状态

转载说明:原创不易,未经授权,谢绝任何形式的转载状态管理是构建任何Web应用程序的重要组成部分。虽然Vue提供了管理简单状态的技术,但随着应用程序复杂性的增加,处理状态可能变得更具挑战性。这就是为什么...

Vue3开发企业级音乐Web App 明星讲师带你学习大厂高质量代码

Vue3开发企业级音乐WebApp明星讲师带你学习大厂高质量代码下栽课》jzit.top/392/...

一篇文章说清 webpack、vite、vue-cli、create-vue 的区别

webpack、vite、vue-cli、create-vue这些都是什么?看着有点晕,不要怕,我们一起来分辨一下。...

超赞 vue2/3 可视化打印设计VuePluginPrint

今天来给大家推荐一款非常不错的Vue可拖拽打印设计器Hiprint。引入使用//main.js中引入安装import{hiPrintPlugin}from'vue-plugin-...

搭建Trae+Vue3的AI开发环境(vue3 ts开发)

从2024年2025年,不断的有各种AI工具会在自媒体中火起来,号称各种效率王炸,而在AI是否会替代打工人的话题中,程序员又首当其冲。...

如何在现有的Vue项目中嵌入 Blazor项目?

...

Vue中mixin怎么理解?(vue的mixins有什么用)

作者:qdmryt转发链接:https://mp.weixin.qq.com/s/JHF3oIGSTnRegpvE6GSZhg前言...

Vue脚手架安装,初始化项目,打包并用Tomcat和Nginx部署

1.创建Vue脚手架#1.在本地文件目录创建my-first-vue文件夹,安装vue-cli脚手架:npminstall-gvue-cli安装过程如下图所示:创建my-first-vue...

新手如何搭建个人网站(小白如何搭建个人网站)

ElementUl是饿了么前端团队推出的桌面端UI框架,具有是简洁、直观、强悍和低学习成本等优势,非常适合初学者使用。因此,本次项目使用ElementUI框架来完成个人博客的主体开发,欢迎大家讨论...

零基础入门vue开发(vue快速入门与实战开发)

上面一节我们已经成功的安装了nodejs,并且配置了npm的全局环境变量,那么这一节我们就来正式的安装vue-cli,然后在webstorm开发者工具里运行我们的vue项目。这一节有两种创建vue项目...

.net core集成vue(.net core集成vue3)

react、angular、vue你更熟悉哪个?下边这个是vue的。要求需要你的计算机安装有o.netcore2.0以上版本onode、webpack、vue-cli、vue(npm...

使用 Vue 脚手架,为什么要学 webpack?(一)

先问大家一个很简单的问题:vueinitwebpackprjectName与vuecreateprojectName有什么区别呢?它们是Vue-cli2和Vue-cli3创建...

vue 构建和部署(vue项目部署服务器)

普通的搭建方式(安装指令)安装Node.js检查node是否已安装,终端输入node-v会使用命令行(安装)npminstallvue-cli-首先安装vue-clivueinitwe...

Vue.js 环境配置(vue的环境搭建)

说明:node.js和vue.js的关系:Node.js是一个基于ChromeV8引擎的JavaScript运行时环境;类比:Java的jvm(虚拟机)...

vue项目完整搭建步骤(vuecli项目搭建)

简介为了让一些不太清楚搭建前端项目的小白,更快上手。今天我将一步一步带领你们进行前端项目的搭建。前端开发中需要用到框架,那vue作为三大框架主流之一,在工作中很常用。所以就以vue为例。...

取消回复欢迎 发表评论: