code: def forward(self, inputs):
      inputs = self.flatten(inputs)
      logits = self.linear_relu_stack(inputs)
      return logits
  
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 2 for calling the NeuralNetwork obj's underlying member method code with example usage: obj.forward(inputs) and returns: logits