深度学习(九)——Tensor基本运算

本文介绍了深度学习中Tensor的基本运算,包括加减乘除、矩阵相乘、次方函数及取整操作。torch.matmul(@)用于矩阵相乘,遵循Pytorch的输出优先原则。对于四维Tensor,特定维度进行矩阵乘法。此外,还讨论了平方根、指数、对数、取整和裁剪等函数的使用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

+:add或者+
-:sub或者-
*:mul或者*
/:div或者/——一个**/符号代表除法,两个//**代表整除
在这里插入图片描述
矩阵相乘
*:是对应位置相乘
torch.mm只适用2d,不推荐
torch.matmul=@:矩阵相乘,@是matmul的重载
在这里插入图片描述
图中w为(512,784),a(4,784)@w(784,512)=(4,512),为什么w的512放第一位呢:Pytorch习惯把输出放第一位,t():转置,适合2d,高维用transpose
在这里插入图片描述
四维*四维:前两位不变,矩阵乘后两位,前两位如果不一致就适用broadcast

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值