import tensorflow as tf

        regressor = XLNetRegressor(hparams=hparams)
        logits = regressor(inputs)

        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            logits_ = sess.run(logits)
            self.assertEqual(logits_.shape,
                             (batch_size, max_time))
