Avoiding looping when using GP prior on latent variables

Questions version agnostic
3

Hm, just tried it. Getting the following error related to tt.repeat call:

File ~/.conda/envs/routines/lib/python3.9/site-packages/theano/tensor/extra_ops.py:786, in repeat(x, repeats, axis)
    783 repeats = basic.as_tensor_variable(repeats)
    785 if repeats.ndim > 1:
--> 786     raise ValueError("The dimension of repeats should not exceed 1.")
    788 if repeats.ndim == 1 and not repeats.broadcastable[0]:
    789     return RepeatOp(axis=axis)(x, repeats)

ValueError: The dimension of repeats should not exceed 1.