code: def nonlinearity(self, x):
      return x - logsumexp(x, axis=1, keepdims=True)
  
imports: from __future__ import absolute_import
  from __future__ import print_function
  from builtins import range
  import autograd.numpy
  import autograd.numpy.random
  import autograd.scipy.signal
  from autograd import grad
  import data_mnist
label_desc: Use image node_id 1 for calling the softmax_layer obj's underlying member method code with example usage: obj.nonlinearity(x) and returns: unknown