SYMBOL INDEX (26 symbols across 3 files)

FILE: capsnet.py
  class ConvLayer (line 9) | class ConvLayer(nn.Module):
    method __init__ (line 10) | def __init__(self, in_channels=1, out_channels=256, kernel_size=9):
    method forward (line 19) | def forward(self, x):
  class PrimaryCaps (line 23) | class PrimaryCaps(nn.Module):
    method __init__ (line 24) | def __init__(self, num_capsules=8, in_channels=256, out_channels=32, k...
    method forward (line 31) | def forward(self, x):
    method squash (line 37) | def squash(self, input_tensor):
  class DigitCaps (line 43) | class DigitCaps(nn.Module):
    method __init__ (line 44) | def __init__(self, num_capsules=10, num_routes=32 * 6 * 6, in_channels...
    method forward (line 53) | def forward(self, x):
    method squash (line 78) | def squash(self, input_tensor):
  class Decoder (line 84) | class Decoder(nn.Module):
    method __init__ (line 85) | def __init__(self, input_width=28, input_height=28, input_channel=1):
    method forward (line 99) | def forward(self, x, data):
  class CapsNet (line 114) | class CapsNet(nn.Module):
    method __init__ (line 115) | def __init__(self, config=None):
    method forward (line 132) | def forward(self, data):
    method loss (line 137) | def loss(self, data, x, target, reconstructions):
    method margin_loss (line 140) | def margin_loss(self, x, labels, size_average=True):
    method reconstruction_loss (line 153) | def reconstruction_loss(self, data, reconstructions):

FILE: data_loader.py
  class Dataset (line 5) | class Dataset:
    method __init__ (line 6) | def __init__(self, dataset, _batch_size):

FILE: test_capsnet.py
  class Config (line 22) | class Config:
    method __init__ (line 23) | def __init__(self, dataset='mnist'):
  function train (line 74) | def train(model, optimizer, train_loader, epoch):
  function test (line 107) | def test(capsule_net, test_loader, epoch):