Coregionalize kernel example?

Does anyone have a working example of using the Coregionalize kernel in Pyro (preferably one with more than two components)? I’m unable to get a model that uses one up and running, so it would he helpful to see it. Specifically, I can’t seem to get the shapes of the inputs right. I’ve looked at the test suite example, but it has not been helpful.