What's wrong with transmission?

Since it appears to resolve your concern to shut off and restart your vehicle, I would suspect some kind of "input" issue to the engine and transmission controls. Test equipment should be utilized to monitor transmission operation and the transmission position sensor while driving and while the problem occurs to see what may be failing.