Đạo hàm tự động với autograd
DễPyTorchautograd4 test · 20 giây · sai số 1e-4
Dùng autograd của PyTorch để tính đạo hàm của
f(x) = Σ (xᵢ³ − 2xᵢ² + 5xᵢ)
theo từng phần tử của x.
1. Tạo bản sao cần đạo hàm: x = x.clone().requires_grad_(True)
2. Tính f, gọi f.backward()
3. Trả về x.grad
(Kiểm tra tay: ∂f/∂xᵢ = 3xᵢ² − 4xᵢ + 5.)
Viết hàm
poly_grad(x)Tham số
x: tensor (n,)
Trả về
Tensor (n,) — gradient.
Ví dụ
Tham số
x = [0,1,2]
Kết quả
[5.0,4.0,9.0]
Tham số
x = [-1]
Kết quả
[12.0]
Bài tiếp theo: Cross-entropy bằng tensor →
Python · PyTorch
Bấm “Chạy thử” để gọi poly_grad() với 2 ví dụ trong đề và so sánh kết quả. Dùng print() thoải mái để gỡ lỗi.