code: def forward(self, edge_attr):
      bond_embedding = []
      for i in range(edge_attr.shape[1]):
          bond_embedding.append(
              self.bond_embedding_list[i](edge_attr[:, i])
          )
  
      bond_embedding = paddle.concat(bond_embedding, axis=1)
  
      return self.mlp(bond_embedding)
  
imports: import numpy
  import paddle
  import paddle.nn
  import paddle.distributed
  from pgl.utils.logger import log
  import features.local_feature
label_desc: Use image node_id 2 for calling the CatBondEncoder obj's underlying member method code with example usage: obj.forward(edge_attr) and returns: self