从零实现深度学习框架——实现Tensor的反向传播
在常见运算的计算图中,我们了解了加减乘除等运算的计算图。本文通过代码实现加法和乘法的计算图来了解我们的`Tensor`自动反向传播计算梯度的模式。
计算 Python 代码的内存和模型显存消耗的小技巧
本篇文章我们将介绍两个 Python 库 memory_profiler和Pytorch-Memory-Utils,这两个库可以帮助我们了解内存和显存的消耗。
18 个 实用的Numpy 代码片段总结
在本文中,我整理了一些 NumPy 代码的片段,这些代码片段都是在日常开发中经常用到的。