Softmax.py

#! /usr/bin/env python
"""
Author: Laksh Arora
Program: SOFTMAX.PY
Date: Friday, April 15 2016
Description: Simple softmax function.
"""

import numpy as np
npa = np.array

def softmax(w, t = 1.0):
    e = np.exp(npa(w) / t)
    # can handle 2-D array too!
    dist = e / e.sum(axis=0)
    return dist

if __name__ == '__main__':

    w = np.array([1.0, 2.0, 3.0])
    print softmax(w)