Auto weight init., non-square shape fix