code: def __init__(
      self,
      input_dim,
      output_dim,
      hid_dim=64,
      bias=True,
      temperature=0.1,
  ):
      super(ST_Dis, self).__init__()
  
      self.input_dim = input_dim
      self.hid_dim = hid_dim
      self.output_dim = output_dim
      self.bias = bias
      self.temperature = temperature
  
      self.linear1 = nn.Linear(input_dim, hid_dim, bias=bias)
      self.linear2 = nn.Linear(hid_dim, output_dim, bias=bias)
      self.tanh = nn.Tanh()
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from .df_utils import one_hot_argmax
label_desc: Use image node_id 1 to create a new ST_Dis object from inherited base classes: nn with example: obj = ST_Dis(input_dim, output_dim, hid_dim, bias, temperature)