China has urged the US to drop its "Cold War mentality" after Washington said it planned to diversify its nuclear armoury with smaller bombs.