How to set up sigma for multiple batches of data?

oh. I figured out the error was in the line:

@as_op(itypes=[pt.dvector], otypes=[pt.dmatrix])

I used otypes=[pt.dtensor3] and it seems to be working now. I’m using the shape=(6,11) for sigmas.

Thank you very much.

1 Like