🧊 又挖到一份 PyTorch 入门笔记,专治「张量这块一直没搞明白」
GitHub 上 1K stars
学深度学习卡在第一关的人比想象中多。教程一上来就是搭网络训模型,但真到自己写的时候,形状对不上、广播规则不清楚、矩阵乘法维度顺序搞反,报错信息看半天不知道错在哪。问题不在模型,在张量这层基本功没打牢。
这份材料就只干这一件事,把张量从头讲透:张量怎么初始化、类型怎么转、数学运算和比较运算怎么用、矩阵乘法和批处理怎么对齐维度、广播规则到底怎么生效、索引和形状变换有哪些写法。
形式是一个 Jupyter Notebook,`pytorch_fundamentals.ipynb`,边看边改边跑,比读文档快得多。MIT 协议。
地基没打好,上面盖多高都会晃。
GitHub:
顯示更多