code: def __init__(self, in_channels, hidden_channels, out_channels):
      super().__init__()
  
      self.conv1 = ARMAConv(
          in_channels,
          hidden_channels,
          num_stacks=3,
          num_layers=2,
          shared_weights=True,
          dropout=0.25,
      )
  
      self.conv2 = ARMAConv(
          hidden_channels,
          out_channels,
          num_stacks=3,
          num_layers=2,
          shared_weights=True,
          dropout=0.25,
          act=lambda x: x,
      )
  
file_path: pytorch_geometric/examples/arma.py
imports: import os.path
  import torch
  import torch.nn.functional
  import torch_geometric.transforms
  from torch_geometric.datasets import Planetoid
  from torch_geometric.nn import ARMAConv
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: torch.nn with example: obj = Net(in_channels, hidden_channels, out_channels)