Using PyTensor.scan to loop over multidimensional array

Questions v5
12

Yes that works, but I need to use:

A_sim = A_sim.dimshuffle('x',1, 0)
A_new = pt.concatenate([A_sim, result], axis=0)

because A_sim is a subtensor of A_array and has less dimensions.

1 Like