Hi Kevin
I had exactly the same issue with my A4 B8. Had it diagnostically checked out but no issues showed up. I was told to take it to ann auto gearbox specialist. They took it aprt and found one of the plates ( can't remember which one ) was looking a bit suspect. They replaced it but only marginal improvement.