Code Explanation:
Line 1
import torch
This imports the PyTorch library.
PyTorch is a powerful library for tensor computations and automatic differentiation, often used in deep learning.
Line 2
x = torch.tensor(2.0, requires_grad=True)
Creates a tensor x with the value 2.0.
requires_grad=True tells PyTorch:
“Please keep track of all operations involving this tensor.”
So later, we can calculate gradients (i.e., derivatives) with respect to x.
Line 3
y = x**3 + 2 * x + 1
Defines a function y in terms of x:
Since x has requires_grad=True, PyTorch builds a computation graph behind the scenes.
Every operation (**3, *2, +1) is tracked so we can differentiate y later.
Line 4
y.backward()
This tells PyTorch to compute the derivative of y with respect to x.
Since y is a scalar (a single value), calling .backward() automatically computes:
Line 5
print(x.grad)
Prints the computed gradient of y with respect to x.
Final Output:
tensor(14.)
.png)

0 Comments:
Post a Comment