Extend log_softmax stabilization rewrite to graphs with indexing and expand_dims#352

ricardoV94 merged 2 commits into

pymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:stabilize_log_subtensor_softmaxricardoV94/pytensor:stabilize_log_subtensor_softmaxCopy head branch name to clipboard