class_docstr: Implementation of set2set pooling operator.
  
  This is an implementation of the paper ORDER MATTERS: SEQUENCE TO SEQUENCE 
  FOR SETS (https://arxiv.org/pdf/1511.06391.pdf).
class_docstr_tok: ["Implementation","of","set2set","pooling","operator",".","This","is","an","implementation","of","the","paper","ORDER","MATTERS",":","SEQUENCE","TO","SEQUENCE","FOR","SETS","(","https",":","\/\/arxiv.org\/pdf\/1511.06391.pdf",")","."]
docstr: Args:
      input_dim: hidden size of input data.
      n_iters: number of set2set iterations.
      n_layers: number of lstm layers.
docstr_tok: ["Args",":","input_dim",":","hidden","size","of","input","data",".","n_iters",":","number","of","set2set","iterations",".","n_layers",":","number","of","lstm","layers","."]
code: def __init__(self, input_dim, n_iters, n_layers):
      """
      Args:
          input_dim: hidden size of input data.
          n_iters: number of set2set iterations.
          n_layers: number of lstm layers.
      """
      self.input_dim = input_dim
      self.output_dim = 2 * input_dim
      self.n_iters = n_iters
  
      # this's set2set n_layers, lstm n_layers = 1
      self.n_layers = n_layers
  
file_path: PGL/legacy/pgl/layers/set2set.py
imports: from __future__ import division
  from __future__ import absolute_import
  from __future__ import print_function
  from __future__ import unicode_literals
  import paddle.fluid
  import paddle.fluid.layers
  import pgl
label_desc: Use image node_id 1 to create a new Set2Set object from inherited base classes: object with example: obj = Set2Set(input_dim, n_iters, n_layers)