code: def forward(
      self, *args: pytorch.TorchData, **kwargs: Any
  ) -> torch.Tensor:
      assert len(args) == 1
      x = args[0]
      assert isinstance(x, torch.Tensor)
      return x.contiguous().view(x.size(0), -1)
  
file_path: determined/examples/tutorials/mnist_pytorch/model.py
imports: from typing import Any, Dict
  import torch
  from torch import nn
  from determined import pytorch
label_desc: Use image node_id 1 for calling the Flatten obj's underlying member method code with example usage: obj.forward() and returns: x