code: def softmax(x, temperature):
      """Compute softmax values for each sets of scores in x."""
      x = x / temperature
      return torch.nn.functional.softmax(x, dim=1)
  
file_path: ReAgent/reagent/core/torch_utils.py
imports: from io import BytesIO
  from typing import Dict, List
  import numpy
  import torch
  from torchrec.sparse.jagged_tensor import JaggedTensor, KeyedJaggedTensor
label_desc: Use image node_id 5 for calling a global function with example usage: softmax(x, temperature) and returns: torch