aroraakshit

import numpy as np

def softmax(x):
  """ Compute the softmax for each row of the input x
  Arguments:
  x -- A N dimensional veector or M X N dimensional numpy matrix.
  Return:
  x -- modified x in-place
  """