SYMBOL INDEX (39 symbols across 5 files)

FILE: tests/test_unet.py
  function run (line 11) | def run(model, shape):
  function test_unet_1d (line 18) | def test_unet_1d():
  function test_unet_1d_residual (line 30) | def test_unet_1d_residual():
  function test_unet_2d (line 42) | def test_unet_2d():
  function test_unet_2d_residual (line 54) | def test_unet_2d_residual():
  function test_unet_3d (line 66) | def test_unet_3d():
  function test_unet_3d_residual (line 78) | def test_unet_3d_residual():

FILE: unet/conv.py
  class ConvolutionalBlock (line 6) | class ConvolutionalBlock(nn.Module):
    method __init__ (line 7) | def __init__(
    method forward (line 77) | def forward(self, x):
    method add_if_not_none (line 81) | def add_if_not_none(module_list, module):

FILE: unet/decoding.py
  class Decoder (line 19) | class Decoder(nn.Module):
    method __init__ (line 20) | def __init__(
    method forward (line 58) | def forward(self, skip_connections, x):
  class DecodingBlock (line 65) | class DecodingBlock(nn.Module):
    method __init__ (line 66) | def __init__(
    method forward (line 129) | def forward(self, skip_connection, x):
    method center_crop (line 143) | def center_crop(self, skip_connection, x):
  function get_upsampling_layer (line 155) | def get_upsampling_layer(upsampling_type: str) -> nn.Upsample:
  function get_conv_transpose_layer (line 168) | def get_conv_transpose_layer(dimensions, in_channels, out_channels):
  function fix_upsampling_type (line 175) | def fix_upsampling_type(upsampling_type: str, dimensions: int):

FILE: unet/encoding.py
  class Encoder (line 6) | class Encoder(nn.Module):
    method __init__ (line 7) | def __init__(
    method forward (line 55) | def forward(self, x):
    method out_channels (line 63) | def out_channels(self):
  class EncodingBlock (line 67) | class EncodingBlock(nn.Module):
    method __init__ (line 68) | def __init__(
    method forward (line 141) | def forward(self, x):
    method out_channels (line 158) | def out_channels(self):
  function get_downsampling_layer (line 162) | def get_downsampling_layer(

FILE: unet/unet.py
  class UNet (line 14) | class UNet(nn.Module):
    method __init__ (line 15) | def __init__(
    method forward (line 122) | def forward(self, x):
  class UNet1D (line 131) | class UNet1D(UNet):
    method __init__ (line 132) | def __init__(self, *args, **user_kwargs):
  class UNet2D (line 141) | class UNet2D(UNet):
    method __init__ (line 142) | def __init__(self, *args, **user_kwargs):
  class UNet3D (line 151) | class UNet3D(UNet):
    method __init__ (line 152) | def __init__(self, *args, **user_kwargs):