code: def __init__(self, device="cpu"):
      super(Net, self).__init__()
      self.device = device
      self.conv1 = nn.Sequential(
          nn.Conv2d(
              in_channels=1, out_channels=256, kernel_size=9, stride=1
          ),
          nn.ReLU(inplace=True),
      )
  
      self.primary = PrimaryCapsuleLayer(device=device)
      self.digits = DGLDigitCapsuleLayer(device=device)
  
file_path: dgl/examples/pytorch/capsule/model.py
imports: import torch
  from DGLDigitCapsule import DGLDigitCapsuleLayer
  from DGLRoutingLayer import squash
  from torch import nn
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: nn with example: obj = Net(device)