05动手学深度学习(下)

按特定轴求和

看到后面发现沐神单独分开是为了解释其中的线性代数部分中按特定轴求和部分的理解。

按特定轴求和

python 复制代码
import torch

a=torch.ones((2,5,4))
a.shape
复制代码
torch.Size([2, 5, 4])
python 复制代码
a.sum(axis=[0,2])
复制代码
tensor([8., 8., 8., 8., 8.])
python 复制代码
a.sum(axis=[0,2],keepdims=True).shape
复制代码
torch.Size([1, 5, 1])

上面的都可以换,比如axis可以换成0、1什么的,大家都可以尝试一下,就比较好理解具体是怎样求和的。

相关推荐
丁学文武10 分钟前
大模型原理与实践:第二章-Transformer架构_第2部分Encoder-Decoder架构
人工智能·深度学习·transformer
liliangcsdn22 分钟前
基于ollama运行27b gemma3解决ffmpeg命令生成问题
人工智能·ffmpeg
云雾J视界32 分钟前
算法偏见的解药:将敏捷“灵魂”注入AI伦理
人工智能·算法·谷歌·伦理债·算法偏见·高效程序员·ai决策系统
码界奇点1 小时前
京东JoyAgent-JDGenie开源多智能体系统如何重塑AI应用落地新范式
人工智能·ai·智能手机·开源
小Tomkk2 小时前
AI 提效:利用 AI 从前端 快速转型为UI/UX设计师和产品
前端·人工智能·ui
王哥儿聊AI2 小时前
CompLLM 来了:长文本 Q&A 效率革命,线性复杂度 + 缓存复用,推理速度与效果双丰收
人工智能·深度学习·机器学习·语言模型
minhuan2 小时前
构建AI智能体:四十六、Codebuddy MCP 实践:用高德地图搭建旅游攻略系统
人工智能·mcp·codebuddy·高德api
青云交2 小时前
Java 大视界 -- Java 大数据在智能安防视频监控系统中的视频语义理解与智能检索进阶
java·深度学习·监控系统·行为识别·智能安防·智能检索·视频语义理解
不当菜鸡的程序媛3 小时前
https://duoke360.com/post/35063
人工智能
IT_陈寒3 小时前
SpringBoot3踩坑实录:一个@Async注解让我多扛了5000QPS
前端·人工智能·后端