import tensorflow as tf


    if is_fm_loss:
      global_pool_head = self.end_points_D['global_pool']
      real_data_features = tf.slice(global_pool_head, [0, 0], [batch_size_train, num_classes])
      fake_data_features = tf.slice(global_pool_head, [batch_size_train, 0],
                                    [batch_size_train, num_classes])
