code: def forward(self, x):
      out_conv1 = self.conv1(x)
      out_primary_caps = self.primary(out_conv1)
      out_digit_caps = self.digits(out_primary_caps)
      return out_digit_caps
  
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 2 for calling the Net obj's underlying member method code with example usage: obj.forward(x) and returns: out_digit_caps