code: def softmax(dist):
      """
      Computes the softmax function on a set of values
      """
  
      output = dist - dist.max(axis=0)
      output = np.exp(output)
      output = output / np.sum(output, axis=0)
      return output
  
imports: import numpy
  from scipy import special
  from pymdp import utils
  from itertools import chain
label_desc: Use image node_id 14 for calling a global function with example usage: softmax(dist) and returns: output