def softmax(x):
#     x = x.clone() - this doesn't work
    nx = tr.zeros(x.shape)

    for i in range(x.shape[1]):
        colmax = x[:, i].max()

        for j in range(x.shape[0]):
            nx[j, i] = tr.exp(x[j, i] - colmax)

        col = nx[:, i]
        col /= col.sum()

    return nx