|View source on GitHub|
Conditional mean of inputs given noisy soft rounded values.
tfc.ops.soft_round_conditional_mean( y, alpha )
Computes g(z) = E[Y | s(Y) + U = z] where s is the soft-rounding function, U is uniform between -0.5 and 0.5 and Y is considered uniform when truncated to the interval [z-0.5, z+0.5].
This is described in Sec. 4.1. in the paper
"Universally Quantized Neural Compression"
Eirikur Agustsson & Lucas Theis
The conditional mean, of same shape as