code: def __init__(
      self,
      in_dim,
      hid_dim,
      out_dim,
      num_stacks,
      num_layers,
      activation=None,
      dropout=0.0,
  ):
      super(ARMA4NC, self).__init__()
  
      self.conv1 = ARMAConv(
          in_dim=in_dim,
          out_dim=hid_dim,
          num_stacks=num_stacks,
          num_layers=num_layers,
          activation=activation,
          dropout=dropout,
      )
  
      self.conv2 = ARMAConv(
          in_dim=hid_dim,
          out_dim=out_dim,
          num_stacks=num_stacks,
          num_layers=num_layers,
          activation=activation,
          dropout=dropout,
      )
  
      self.dropout = nn.Dropout(p=dropout)
  
file_path: dgl/examples/pytorch/arma/model.py
imports: import math
  import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new ARMA4NC object from inherited base classes: nn with example: obj = ARMA4NC(in_dim, hid_dim, out_dim, num_stacks, num_layers, activation, dropout)