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.
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.