Automakers build these sensors with a long lifecycle, the only reason you’d need to test or change the transmission speed sensor would be because it began to malfunction again.

The shifts with the transmission may be related to the speed sensor too, but if it still persists after a new speed sensor, then you might need a transmission rebuild.