This isn’t quite so trivial.  First, we need to be able to turn off the noise for testing.  Second, we need to be able to choose when to resample the noise tensors, since for NoisyNet-A3C it’s supposed to be resampled after every k-step policy rollout.  But yes, it should be possible to implement for other kinds of layers, although defining the forward pass might be challenging for some.