Recently, I found a piece of code written in Tensorflow. It serves to
overriding tf.round()'s gradient function with tf.identity()'s using
gradient_override_map function:
import tensorflow as tf
G = tf.get_default_graph()
def quantize(x, k):
n = float(2**k - 1)
with G.gradient_override_map({"Round": "Identity"}):
return tf.round(x * n) / n
And I want to implement it in Theano but have no idea....So could anyone
provide some advice on that?
--
---
You received this message because you are subscribed to the Google Groups
"theano-users" group.
To unsubscribe from this group and stop receiving emails from it, send an email
to [email protected].
For more options, visit https://groups.google.com/d/optout.
