What exactly is the context, for example in the distributions like MADEMoG? I wanted to use this distrubution instead of StandardNormal for the MaskedAutoregressiveFlow, but I keep getting the error 'NoneType' object has no attribute 'shape' when I want to sample from this flow due to the fact that the first dimension of samples is set to context.shape[0], but I don't specify the context. Could you provide an example on how to fix this, or how an example context should be provided?