Norm constraint refactor

Should we use `pylearn2.utils.wraps` for consistency?