Theano shared and prediction not working as expected

Solved!
Guess this was a previous bug. Updated to 3.7 and all is well. Appologies for not seeing this answer earlier!