def METHOD_NAME(self) -> Tensor:
    F = self.F
    alpha = self.alpha
    n_trials = self.n_trials
    sum_alpha = F.sum(alpha, axis=-1)
    return (
        F.broadcast_div(alpha, sum_alpha.expand_dims(axis=-1)) * n_trials
    )