Learning the Positions in CountSketch