elektronn3.models.vnet module

class elektronn3.models.vnet.ContBatchNorm3d(*args, **kwargs)

Bases: torch.nn.modules.batchnorm.

forward(input)

class elektronn3.models.vnet.DownTransition(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x)

elektronn3.models.vnet.ELUCons(relu, nchan)

class elektronn3.models.vnet.InputTransition(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x)

class elektronn3.models.vnet.LUConv(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x)

class elektronn3.models.vnet.OutputTransition(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x)

class elektronn3.models.vnet.UpTransition(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x, skipx)

class elektronn3.models.vnet.VNet(*args: Any, **kwargs: Any)

Bases: torch.nn.

forward(x)

elektronn3.models.vnet.passthrough(x, **kwargs)