def sample_from_activation(self, vmap): rate = self.rate_from_activation(vmap) return samplers.truncated_exponential(rate)
def sample_from_activation(self, vmap): return samplers.truncated_exponential( -vmap[self]) # lambda = -activation!
def sample_from_activation(self, vmap): return samplers.truncated_exponential(-vmap[self]) # lambda = -activation!