Gumbel-Softmax sampling of z_pres has been decoupled from downstream computation by means of tf.where. Gradients of q(z_pres|x) wrt parameters back-propagate properly now.
Gumbel-Softmax sampling of z_pres has been decoupled from downstream computation by means of tf.where. Gradients of q(z_pres|x) wrt parameters back-propagate properly now.

aakhundov/tf-attend-infer-repeat