View source on GitHub |
Differentiable approximation to round
.
tfc.ops.soft_round(
x, alpha, eps=0.001
)
Larger alphas correspond to closer approximations of the round function. If alpha is close to zero, this function reduces to the identity.
This is described in Sec. 4.1. in the paper
"Universally Quantized Neural Compression"
Eirikur Agustsson & Lucas Theis
https://arxiv.org/abs/2006.09952
Args | |
---|---|
x
|
tf.Tensor . Inputs to the rounding function.
|
alpha
|
Float or tf.Tensor . Controls smoothness of the approximation.
|
eps
|
Float. Threshold below which soft_round will return identity.
|
Returns | |
---|---|
tf.Tensor
|