INNER CODE UNIT · Python
cond
nv-tlabs/ATISS · scene_synthesis/losses/__init__.py:109
cond = (target < -0.999).float()
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)