torch.Tensor.backward — PyTorch 1.10.1 documentation
pytorch.org › generated › torchtorch.Tensor.backward. Tensor.backward(gradient=None, retain_graph=None, create_graph=False, inputs=None)[source] Computes the gradient of current tensor w.r.t. graph leaves. The graph is differentiated using the chain rule. If the tensor is non-scalar (i.e. its data has more than one element) and requires gradient, the function additionally ...