Solved by pumplerod in post #7 
              
              
                Oh, interesting.  Thank you so much.  Looks like it will certainly save a multiplication step.  I’m not sure what the implications on autograd that will have but hopefully some speed improvement, even though it looks like it will still consume memory where it ought not. 
I believe the slight modific…