import tensorflow as tf

        rnd_pred_cri = tf.squeeze(mlp(tf.concat([x_ph, a_ph], axis=-1), list(hidden_sizes) + [1], activation, None),
                                  axis=1)
        rnd_pred_cri_in_ph = tf.concat([x_ph, a_ph], axis=-1)
        rnd_pred_cri_in_dim = rnd_pred_cri_in_ph.shape.as_list()[1]
