博客 / 列表

co_yield - 從0構建深度學習框架——揭秘深度學習框架的黑箱

引言 你有沒有好奇過,當你在 PyTorch 或 TensorFlow 中調用 .backward() 計算梯度時,框架到底在背後做了什麼? 我們每天都在使用這些成熟的深度學習工具,但很少有人真正去探索它們的底層實現——自動微分的魔法、計算圖的構建、張量運算的優化……這些隱藏在API背後的核心原理,才是深度學習的真正基石。 今天,我將向大家介紹一個從零開始實現的輕量級深度學習框架——Needle。

人工智能