weight matrix init with uniform dist