No. It has been on my list of things to do when I have time ever since I implemented einsum, but it never happened.  You’d make a lot of friends if you implemented that. 
The (third party) opt_einsum 99 package supports PyTorch, so you could use that, too.
Best regards
Thomas