ast_toolbox.utils.exp_utils module

ast_toolbox.utils.exp_utils.log_mean_exp(x, dim)

Compute the log(mean(exp(x), dim)) in a numerically stable manner

ast_toolbox.utils.exp_utils.log_sum_exp(x, dim)

Compute the log(sum(exp(x), dim)) in a numerically stable manner

ast_toolbox.utils.exp_utils.softmax(x, dim)

Compute softmax values for each sets of scores in x along dim