code: def __init__(self, non_tensor_output):
      super().__init__(HIDDEN_DIM, HIDDEN_DIM)
      self.non_tensor_output = non_tensor_output
  
imports: import pytest
  import torch
  import deepspeed
  from copy import deepcopy
  from unit.common import DistributedTest
label_desc: Use image node_id 1 to create a new LinearNonTensorOutput object from inherited base classes: torch.nn with example: obj = LinearNonTensorOutput(non_tensor_output)