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什么的,大家都可以尝试一下,就比较好理解具体是怎样求和的。

相关推荐
星期天要睡觉1 小时前
自然语言处理(NLP)——自然语言处理原理、发展历程、核心技术
人工智能·自然语言处理
低音钢琴1 小时前
【人工智能系列:机器学习学习和进阶01】机器学习初学者指南:理解核心算法与应用
人工智能·算法·机器学习
大千AI助手2 小时前
Hoeffding树:数据流挖掘中的高效分类算法详解
人工智能·机器学习·分类·数据挖掘·流数据··hoeffding树
新知图书3 小时前
大模型微调定义与分类
人工智能·大模型应用开发·大模型应用
山烛3 小时前
一文读懂YOLOv4:目标检测领域的技术融合与性能突破
人工智能·yolo·目标检测·计算机视觉·yolov4
大千AI助手3 小时前
独热编码:分类数据处理的基石技术
人工智能·机器学习·分类·数据挖掘·特征工程·one-hot·独热编码
钱彬 (Qian Bin)3 小时前
项目实践4—全球证件智能识别系统(Qt客户端开发+FastAPI后端人工智能服务开发)
人工智能·qt·fastapi
钱彬 (Qian Bin)3 小时前
项目实践3—全球证件智能识别系统(Qt客户端开发+FastAPI后端人工智能服务开发)
人工智能·qt·fastapi
Microsoft Word3 小时前
向量数据库与RAG
数据库·人工智能·向量数据库·rag
2401_836900334 小时前
YOLOv5:目标检测的实用派王者
人工智能·计算机视觉·目标跟踪·yolov5