code: def run_softmax_reference(input):
      return torch.nn.functional.softmax(input, dim=-1)
  
imports: import pytest
  import torch
  import deepspeed
  from deepspeed.ops.op_builder import InferenceBuilder
label_desc: Use image node_id 2 for calling a global function with example usage: run_softmax_reference(input) and returns: torch