Solved by tom in post #2 
              
              
                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 package supports PyTorch, so you could use that, too. 
Best regards 
Thomas