Let numpy infer dimension for slightly more terse code