Einsum in pytensor

Yep, that works perfectly. Many thanks, you just saved me a whole lot of hassle barking up the wrong tree :joy: