tfc.ops.soft_round_inverse

Inverse of soft_round.

tfc.ops.soft_round_inverse(
    y, alpha, eps=0.001
)

This is described in Sec. 4.1. in the paper

"Universally Quantized Neural Compression"
Eirikur Agustsson & Lucas Theis
https://arxiv.org/abs/2006.09952

y tf.Tensor. Inputs to this function.
alpha Float or tf.Tensor. Controls smoothness of the approximation.
eps Float. Threshold below which soft_round is assumed to equal the identity function.

tf.Tensor