PyTorch · Class

TensorDataset

torch.nn is a set of neural-network building blocks. Together with optimizers, datasets, and data loaders, it helps define models, update their weights, and feed them batches.

Last updated 2 November 2026

class MyModel(nn.Module):
    def __init__(self):
        super().__init__()

Source: What is torch.nn really? — PyTorch Tutorials 2.15.0+cu132 documentation. Checked 2026-11-03.