code: def edge_transfer(self, edges):
      # Transform from Bessel basis to dence vector
      rbf = self.dense_rbf(edges.data["rbf"])
      # Initial transformation
      x_ji = self.dense_ji(edges.data["m"])
      x_kj = self.dense_kj(edges.data["m"])
      if self.activation is not None:
          x_ji = self.activation(x_ji)
          x_kj = self.activation(x_kj)
  
      # w: W * e_RBF \bigodot \sigma(W * m + b)
      return {"x_kj": x_kj * rbf, "x_ji": x_ji}
  
file_path: dgl/examples/pytorch/dimenet/modules/interaction_block.py
imports: import dgl.function
  import torch
  import torch.nn
  from modules.initializers import GlorotOrthogonal
  from modules.residual_layer import ResidualLayer
label_desc: Use image node_id 3 for calling the InteractionBlock obj's underlying member method code with example usage: obj.edge_transfer(edges) and returns: dict