INNER CODE UNIT · Python

log_probs

nv-tlabs/ATISS · scene_synthesis/losses/__init__.py:110

    log_probs = cond * log_cdf_plus + (1. - cond) * inner_out

    log_probs = log_probs + F.log_softmax(logit_probs, -1)
    return -log_sum_exp(log_probs)

View source record →