requires_grad
)backward()
).grad
)torch.nn
torch.nn.Module
Base Classtorch.nn
losses)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
.grad
)Was this section helpful?
requires_grad
, backward()
, and how gradients are stored in the .grad
attribute.