您当前的位置:首页 > 电脑百科 > 程序开发 > 算法

PyTorch张量的四种乘法运算

时间:2023-03-21 11:58:44  来源:今日头条  作者:长颈鹿睡觉

在PyTorch中有四种类型的乘法运算(位置乘法、点积、矩阵与向量乘法、矩阵乘法),非常容易搞混,我们一起来看看这四种乘法运算的区别。

位置乘法

先构建两个张量a,b他们都是4行5列。

a = torch.arange(20).reshape([4,5])
b = torch.randn([4,5])

 

位置乘法,顾名思义就是将两个张量对应位置的元素进行乘法运算,运算符是*。

可以是两个张量相乘,也可以是标量和张量相乘。

标量与张量相乘,是用标量与张量的每个元素相乘,结果张量的形状不变。

4 * a

 

两个张量相乘,是对应位置的元素相乘,结果张量的形状不变。

a * b

 

点积

点积是两个向量(也就是一维张量)对应位置的元素相乘后求和,结果是一个标量,使用dot函数进行计算。

先构建两个向量a、b,点积操作要求两个向量的数据类型要一致,因此a中指定数据类型为float。

a = torch.arange(6, dtype=torch.float32)
b = torch.ones(6)

 

执行点积操作,结果是一个标量。

torch.dot(a,b)

 

矩阵与向量乘法

矩阵(二维张量)与向量(一维张量)的乘法是将矩阵的每一行与向量进行点积,要求矩阵的列维数与向量的维数相同,结果的维数与行数相同。

使用mv函数进行运算。

构建一个4行5列的矩阵和一个维数为5的向量。

a = torch.arange(20,dtype=torch.float32).reshape([4,5])
b = torch.ones(5)

 

使用mv函数相乘后,结果是维数为4的向量。

torch.mv(a,b)

 

矩阵乘法

矩阵(二维张量)乘法是用第一个矩阵的行向量与第二个矩阵的列向量进行点积,要求第一个矩阵的列数与第二个矩阵的行数相同。

使用mm函数进行运算。

构建两个矩阵,一个4行5列,一个5行6列

a = torch.arange(20,dtype=torch.float32).reshape([4,5])
b = torch.randn([5,6])

 

使用mm函数相乘后,结果是4行6列的矩阵。

torch.mm(a,b)

 



Tags:PyTorch   点击:()  评论:()
声明:本站部分内容及图片来自互联网,转载是出于传递更多信息之目的,内容观点仅代表作者本人,如有任何标注错误或版权侵犯请与我们联系(Email:2595517585@qq.com),我们将及时更正、删除,谢谢。
▌相关推荐
在PyTorch中有四种类型的乘法运算(位置乘法、点积、矩阵与向量乘法、矩阵乘法),非常容易搞混,我们一起来看看这四种乘法运算的区别。位置乘法先构建两个张量a,b他们都是4行5列。a...【详细内容】
2023-03-21  Tags: PyTorch  点击:(0)  评论:(0)  加入收藏
使用大型数据集训练大型深度神经网络 (DNN) 的问题是深度学习领域的主要挑战。 随着 DNN 和数据集规模的增加,训练这些模型的计算和内存需求也会增加。 这使得在计算资源有限...【详细内容】
2023-02-19  Tags: PyTorch  点击:(29)  评论:(0)  加入收藏
PyTorch机器学习框架遭遇恶意依赖供应链链攻击。与Keras 和 TensorFlow类似,PyTorch是基于Python的开源机器学习框架。PyTorch广泛用于计算机视觉和自然语言处理任务。12月31...【详细内容】
2023-01-08  Tags: PyTorch  点击:(34)  评论:(0)  加入收藏
Alex 羿阁 发自 凹非寺 量子位 | 公众号 QbitAI “所有在2022年12月25日至12月30日期间,在Linux系统安装了PyTorch-nightly(每日更新版)的用户,请立即卸载!”上述消息来自PyTorch...【详细内容】
2023-01-03  Tags: PyTorch  点击:(66)  评论:(0)  加入收藏
众所周知,经历了计算机科学过去几十年没有突破的瓶颈期之后,人工智能技术这些年由于算法和算力(并行计算)革命,迈入了一个高速发展的时期。不仅如此,在编程语言和开发者环境不断演...【详细内容】
2022-12-01  Tags: PyTorch  点击:(76)  评论:(0)  加入收藏
学习目标 了解分类器的任务和数据样式 掌握如何用Pytorch实现一个分类器分类器任务和数据介绍 构造一个将不同图像进行分类的神经网络分类器, 对输入的图片进行判别并完成分...【详细内容】
2022-10-25  Tags: PyTorch  点击:(64)  评论:(0)  加入收藏
机器之心报道编辑:杜伟 想要了解自己的 PyTorch 项目在哪些地方分配 GPU 内存以及为什么用完吗?不妨试试这个可视化工具。 近日,PyTorch 核心开发者和 FAIR 研究者 Zachary DeV...【详细内容】
2022-10-24  Tags: PyTorch  点击:(104)  评论:(0)  加入收藏
作者 | 锦恢@知乎 (已授权)来源 | https://zhuanlan.zhihu.com/p/220403674编辑 | 极市平台一、网络结构的可视化我们训练神经网络时,除了随着step或者epoch观察损失函数的走势...【详细内容】
2022-09-19  Tags: PyTorch  点击:(163)  评论:(0)  加入收藏
出品 | OSC开源社区(ID:oschina2013)在 Meta (原 Facebook)、AMD、AWS、谷歌云、微软和 NVIDIA 等大厂的联合推动下,PyTorch 基金会成立了。PyTorch 基金会隶属于 Linux 基金会...【详细内容】
2022-09-13  Tags: PyTorch  点击:(88)  评论:(0)  加入收藏
本文分享自华为云社区《AnimeGANv2 照片动漫化:如何基于 PyTorch 和神经网络给 GirlFriend 制作漫画风头像?【秋招特训】-云社区-华为云》,作者:白鹿第一帅 。前言将现实世界场...【详细内容】
2022-09-13  Tags: PyTorch  点击:(128)  评论:(0)  加入收藏
▌简易百科推荐
在PyTorch中有四种类型的乘法运算(位置乘法、点积、矩阵与向量乘法、矩阵乘法),非常容易搞混,我们一起来看看这四种乘法运算的区别。位置乘法先构建两个张量a,b他们都是4行5列。a...【详细内容】
2023-03-21  长颈鹿睡觉  今日头条  Tags:PyTorch   点击:(0)  评论:(0)  加入收藏
大家好,我是前端西瓜哥。今天来讲讲几何算法中,比较经典的算法:矩形碰撞和包含检测算法。矩形碰撞检测是被广泛使用的算法。比如在游戏中,为了优化图形碰撞判断效率(复杂不规则...【详细内容】
2023-03-13   前端西瓜哥  微信公众号  Tags:算法   点击:(7)  评论:(0)  加入收藏
一、概述三种时间复杂度是O(n)的线性排序算法:桶排序、计数排序、基数排序。二、相似点这三种排序算法都利用了桶的概念,但对桶的使用方法上有明显差异: 基数排序:根据键值的每...【详细内容】
2023-03-13  日拱一卒程序猿  今日头条  Tags:算法   点击:(6)  评论:(0)  加入收藏
SnowFlake 算法,是 Twitter 开源的分布式 ID 生成算法。其核心思想就是:使用一个 64 bit 的 long 型的数字作为全局唯一 ID。在分布式系统中的应用十分广泛,且 ID 引入了时间戳...【详细内容】
2023-03-09  雨夜青草  CSDN  Tags:雪花算法   点击:(16)  评论:(0)  加入收藏
很多时候需要用到连续的id进行数据对比,如判断是否连续等问题。那么,生成连续整数的方式有多种,首先容易想到的是逐步循环,如果想生成1kw条记录,则需要循环1kw次进行插入,那么有没...【详细内容】
2023-03-09  数据库干货铺  今日头条  Tags:整数   点击:(15)  评论:(0)  加入收藏
如何确定桶的 区间范围,有很多种不同的方式。我们这里创建的桶数量等于原始数列的元素数量,除最后一个桶只包含数列最大值外, 前面各个桶的区间按照比例来确定。一、定义桶排序...【详细内容】
2023-03-09  日拱一卒程序猿  今日头条  Tags:桶排序   点击:(10)  评论:(0)  加入收藏
一、推荐业务简介首先介绍一下阿里健康的业务背景和对现状的分析。1、推荐场景展示可解释性推荐,举例而言,如下图中当当网“根据您浏览的商品推荐”(告诉用户推荐的理由)及淘宝...【详细内容】
2023-03-08  DataFunTalk    Tags:算法   点击:(11)  评论:(0)  加入收藏
常见分布式协议和算法的说明和对比开发分布式系统最关键的就是根据场景特点,选择合适的算法,在一致性和可用性之间妥协折中,而妥协折中的关键就在于能否理解各算法的特点。分布...【详细内容】
2023-03-08  后端系统和架构  微信公众号  Tags:算法   点击:(10)  评论:(0)  加入收藏
一、定义同冒泡排序一样,快速排序也属于交换排序,通过元素之间的比较和交换位置来达到排序的目的。不同的是,冒泡排序在每一轮中只把1个元素冒泡到数列的一端,而快速排序则在每...【详细内容】
2023-03-07    今日头条  Tags:快速排序   点击:(8)  评论:(0)  加入收藏
时间序列分解是时序分析中的重要方法,广泛应用于时间序列预测,时间序列异常检测,时间序列聚类等场景,在工业界有很多的落地应用。一个时间序列往往是以下几类变化形式的叠加或耦...【详细内容】
2023-03-07  人工智能技术  微信公众号  Tags:序列   点击:(18)  评论:(0)  加入收藏
站内最新
站内热门
站内头条