Modelling the non-uniform sediment transport by means of the statistical moment approach