Imports

First, import the required modules:

1import torch
2from torch import nn

Then define the model:

10class Model(nn.Module):
11    def __init__(self):
12        super().__init__()