← 全部标签

标签:PyTorch

共 9 篇相关内容,按发布时间排列。

LLaMA(11):用代码验证 RoPE 的远程衰减

修正并扩展 ROPE.ipynb 的实验,用位置核曲线、平移不变测试和距离分桶统计理解 RoPE 的振荡与远程衰减倾向。

LLaMA(10):用 PyTorch 从零实现 RoPE

以课程 ROPE.ipynb 为主线,完成频率预计算、复数旋转、广播与 shape 检查,并验证范数和相对位置性质。

LLaMA(5):从零实现 RMSNorm

以课程 RMSNorm.ipynb 为代码主线,从最小实现出发补齐 epsilon、混合精度、正确验证和 PyTorch 对照测试。

Ascend C Kernel 接入 PyTorch:pybind11 与 torch.library

对比两条 PyTorch Kernel 直调路径,梳理 C++ 封装、NPU Stream、动态库构建和 Python 调用。

深度学习:LSTM 长短期记忆网络

理解 LSTM 为什么能缓解普通 RNN 的长距离依赖问题,掌握细胞状态、遗忘门、输入门、输出门、候选记忆的计算逻辑,以及 PyTorch 中 nn.LSTM 的输入输出 shape。

Transformer(三):多头注意力与张量变形

理解多头注意力为什么有效,逐步追踪拆头、并行计算、拼接与输出投影的张量形状,并给出可用于交叉注意力的实现。

Transformer(八):基于 Notebook 的手写组装与代码调试

以课程 Notebook 的 ModelArgs、MultiHeadAttention、Encoder、Decoder 与 Transformer 类为主线,完成代码整合、错误修正和最小测试。

深度学习:手写数字识别 — 从零手动实现 vs PyTorch 框架实现

基于 MNIST 数据集,先手动实现一个完整的多分类神经网络(前向/反向/参数更新),再用 PyTorch 的 nn.Module/CrossEntropyLoss/SGD 重写,对比两者理解框架帮我们做了什么

LSTM 股价预测笔记:原理与 PyTorch 实现

LSTM 的门控机制推导、股价预测中的数据构造细节,以及一个最小可复现的 PyTorch 实现。