original_docstring: 
          Forward Propagation of DeepCaps Model.
          
docstring: Forward Propagation of DeepCaps Model.
docstring_tokens: Forward Propagation of DeepCaps Model .
code: def forward(self, x, target=None):
          '''
          Forward Propagation of DeepCaps Model.
          '''
  
          x = self.conv1(x)
          x = self.bn1(x)
          x = self.toCaps(x)
  
          x = self.conv2dcaps_00(x)
          x_skip = self.conv2dcaps_01(x)
          x = self.conv2dcaps_02(x)
          x = self.conv2dcaps_03(x)
  
          x = x + x_skip
  
          x = self.conv2dcaps_10(x)
          x_skip = self.conv2dcaps_11(x)
          x = self.conv2dcaps_12(x)
          x = self.conv2dcaps_13(x)
  
          x = x + x_skip
  
          x = self.conv2dcaps_20(x)
          x_skip = self.conv2dcaps_21(x)
          x = self.conv2dcaps_22(x)
          x = self.conv2dcaps_23(x)
  
          x = x + x_skip
          x1 = x
  
          x = self.conv2dcaps_30(x)
          x_skip = self.conv3dcaps_31(x)
          x = self.conv2dcaps_32(x)
          x = self.conv2dcaps_33(x)
  
          x = x + x_skip
          x2 = x
  
          xa = self.flatten_caps(x1)
          xb = self.flatten_caps(x2)
  
          x = torch.cat((xa, xb), dim=-2)
          dig_caps = self.fc_caps(x)
  
          x = self.to_scalar(dig_caps)
  
          masked, indices = self.mask(dig_caps, target)
          decoded = self.decoder(masked)
  
          return dig_caps, masked, decoded, indices
code_tokens: def forward ( self , x , target = None ) : x = self . conv1 ( x ) x = self . bn1 ( x ) x = self . toCaps ( x ) x = self . conv2dcaps_00 ( x ) x_skip = self . conv2dcaps_01 ( x ) x = self . conv2dcaps_02 ( x ) x = self . conv2dcaps_03 ( x ) x = x + x_skip x = self . conv2dcaps_10 ( x ) x_skip = self . conv2dcaps_11 ( x ) x = self . conv2dcaps_12 ( x ) x = self . conv2dcaps_13 ( x ) x = x + x_skip x = self . conv2dcaps_20 ( x ) x_skip = self . conv2dcaps_21 ( x ) x = self . conv2dcaps_22 ( x ) x = self . conv2dcaps_23 ( x ) x = x + x_skip x1 = x x = self . conv2dcaps_30 ( x ) x_skip = self . conv3dcaps_31 ( x ) x = self . conv2dcaps_32 ( x ) x = self . conv2dcaps_33 ( x ) x = x + x_skip x2 = x xa = self . flatten_caps ( x1 ) xb = self . flatten_caps ( x2 ) x = torch . cat ( ( xa , xb ) , dim = - 2 ) dig_caps = self . fc_caps ( x ) x = self . to_scalar ( dig_caps ) masked , indices = self . mask ( dig_caps , target ) decoded = self . decoder ( masked ) return dig_caps , masked , decoded , indices
short_docstring: Forward Propagation of DeepCaps Model.
short_docstring_tokens: Forward Propagation of DeepCaps Model .
comment: '''
          Forward Propagation of DeepCaps Model.
          '''