Method activate_mean

model/mixture_model_utils.py:20–24

(self, means, mean_activation='tanh')

        self.perform_sampling = True
    
    def activate_mean(self, means, mean_activation='tanh'):
        if mean_activation == 'tanh':
            return torch.tanh(means) * (1.0 - eps)  # (-1, 1)
        else:
            raise ValueError(f"Unknown activation function: {self.mean_activation}")
    
    def expand_params(self, logits, means, log_scales, mean_activation='tanh'):
        """