INNER CODE UNIT · Python
inner_out
nv-tlabs/ATISS · scene_synthesis/losses/__init__.py:107
inner_out = inner_cond * log_one_minus_cdf_min + \
(1. - inner_cond) * inner_inner_out
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)