trainnetwork for mixture density network

I want to use the function trainnetwork for mixture density network (MDN), but I always get the error message which is the dimension of ouput of the last layer is mismatch with that of YTrain, I know the reason is the output of MDN includes the mean, the variance and the weight, but I can't get the resolves, who can help me? Thanks a lot.