code: def collect_representation(representation, positions):
      """
      Collect_representation.
  
      :param representation: [batch_size, node_num, feature_dim]
      :param positions: [batch_size, neighbour_num]
      :return: [batch_size, neighbour_num]?
      """
      return collect_probs(representation, positions)
  
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 2 for calling a global function with example usage: collect_representation(representation, positions) and returns: collect_probs