chain rule

The backward() function uses PyTorch's automatic differentiation package, torch.autograd, to differentiate and compute gradients of tensors based on the chain rule.