PyTorch 张量与自动求导 — 练习

Download exercise.py

实现 mse_and_grad(x)：(L=\mathrm{mean}(x^2))，返回 (loss, ∂L/∂x)。对 (x_i) 的梯度是 (2x_i/n)。

cd docs/programming/pytorch/code
python exercise.py

源码位置

docs/programming/pytorch/code/exercise.py