Describe Pyro tensor shapes in the addendum of XOR-MLP example