code: def cal_max_question_representation(reps_lt, attn_scores):
      """
      Calculate max_question_representation.
  
      :param reps_lt: [batch_size, passage_len, hidden_size]
      :param attn_scores: []
      :return: [batch_size, passage_len, hidden_size].
      """
      attn_positions = tf.argmax(attn_scores, axis=2)
      max_reps_lt = collect_representation(reps_lt, attn_positions)
      return max_reps_lt
  
imports: import tensorflow
  from keras import backend
  from keras.engine import Layer
  from matchzoo.contrib.layers.attention_layer import AttentionLayer
label_desc: Use image node_id 1 for calling a global function with example usage: cal_max_question_representation(reps_lt, attn_scores) and returns: max_reps_lt