Dirichlet with random shape

@junpenglao Sounds good. I have opened a new post here, thanks!