geffy

import numpy as np
%pylab inline


def generate_raw_weights(n, a, b, p):
    res = np.zeros((n, n))
    for i in range(n):
        for j in range(n):
            sq_dist = ((i - a)**2 + (j - b)**2) *1.0/ n**2
            res[i,j] = exp(-sq_dist*p)