Variational inference over cartesian product of large sets of observations

Oh I think I get what you mean - since a and b is too large to do dot product, you want to minibatch them to just get a slice… I am not sure it is the right way to do: I dont think dot product itself is batchable