$U(n)$-invariant Kähler–Ricci flow with non-negative curvature