def mean_field_from_activation(self, vmap): rate = self.rate_from_activation(vmap) return samplers.truncated_exponential_mean(rate)
def mean_field_from_activation(self, vmap): return samplers.truncated_exponential_mean(-vmap[self])