split `get_pert_loss` function s.t. we can access perturbed activations if we want to

[jettjaniak]

this can still be a single function, but it should call a function that computes the perturbed activations