One hidden layer (with 64 units and ReLU non-linearity) has been added between the output of RNN and scale, shift, and z_pres generation (separately for each of the three cases).
One hidden layer (with 64 units and ReLU non-linearity) has been added between the output of RNN and scale, shift, and z_pres generation (separately for each of the three cases).

aakhundov/tf-attend-infer-repeat