import tensorflow as tf

        processed_l1_h2 = tf.nn.bias_add(
            processed_l1_h2,
            getattr(self, 'ff_bias_%s' % idx))
        processed_l1_h2 = self.ff_nl(processed_l1_h2)
        if self.batch_norm:
            with tf.variable_scope(
                    'l1_h2_bn_ff_%s' % idx,
                     reuse=self.scope_reuse) as scope:
                processed_l1_h2 = tf.contrib.layers.batch_norm(
                    inputs=processed_l1_h2,
                    scale=True,
                    center=True,
                    fused=True,
                    renorm=False,
                    param_initializers=self.param_initializer,
                    updates_collections=None,
