Function custom_softmax

cnn_class2/test_softmax.py:13–17

(x)

import keras.backend as K

def custom_softmax(x):
  m = tf.reduce_max(x, 1)
  x = x - m
  e = tf.exp(x)
  return e / tf.reduce_sum(e, -1)


a = np.random.randn(1, 1000)