code: def __init__(self):
      super().__init__()
      self.flatten = nn.Flatten()
      self.linear_relu_stack = nn.Sequential(
          nn.Linear(28 * 28, 512),
          nn.ReLU(),
          nn.Linear(512, 512),
          nn.ReLU(),
          nn.Linear(512, 10),
      )
  
file_path: ray/python/ray/train/examples/pytorch/torch_quick_start.py
imports: import torch
  import torch.nn
  from torch.utils.data import DataLoader
  from torchvision import datasets
  from torchvision.transforms import ToTensor
  import ray.train.torch
label_desc: Use image node_id 1 to create a new NeuralNetwork object from inherited base classes: nn with example: obj = NeuralNetwork()