If you would like to add it randomly, you could specify a probability inside the transformation and pass this probability while instantiating it.
On the other hand, if you would like to apply it just for specified data indices, you might need to apply the noise inside the training loop and use the data index (return the index additionally in your Dataset).