def sample_from_activation(self, vmap): p = activation_functions.sigmoid(vmap[self] - vmap[self.flipped_units]) return samplers.bernoulli(p)
def sample_from_activation(self, vmap): p = self.success_probability_from_activation(vmap) return samplers.bernoulli(p)