code: def __init__(self):
      super(LeNet, self).__init__()
      self.conv1 = nn.Conv2d(3, 16, 5)
      self.pool1 = nn.MaxPool2d(2, 2)
      self.conv2 = nn.Conv2d(16, 32, 5)
      self.pool2 = nn.MaxPool2d(2, 2)
      self.fc1 = nn.Linear(32 * 5 * 5, 120)
      self.fc2 = nn.Linear(120, 84)
      self.fc3 = nn.Linear(84, 10)
  
imports: import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new LeNet object from inherited base classes: nn with example: obj = LeNet()