make pylearn2.linear support partial_sum