code: def forward(self, graph_embed):
      loc = self.linear2(self.tanh(self.linear1(graph_embed)))
      loc = one_hot_argmax(loc, self.temperature)
      return loc
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from .df_utils import one_hot_argmax
label_desc: Use image node_id 3 for calling the ST_Dis obj's underlying member method code with example usage: obj.forward(graph_embed) and returns: loc