elektronn3.models.vnet module

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

Bases: _BatchNorm

forward(input)[source]
class elektronn3.models.vnet.DownTransition(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
elektronn3.models.vnet.ELUCons(relu, nchan)[source]
class elektronn3.models.vnet.InputTransition(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
class elektronn3.models.vnet.LUConv(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
class elektronn3.models.vnet.OutputTransition(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
class elektronn3.models.vnet.UpTransition(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, skipx)[source]
class elektronn3.models.vnet.VNet(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
elektronn3.models.vnet.passthrough(x, **kwargs)[source]