elektronn3.models.simple module

class elektronn3.models.simple.Conv3DLayer(*args: Any, **kwargs: Any)

Bases: Module

forward(x)

class elektronn3.models.simple.Extended3DNet(*args: Any, **kwargs: Any)

Bases: Module

forward(x)

class elektronn3.models.simple.N3DNet(*args: Any, **kwargs: Any)

Bases: Module

forward(x)

class elektronn3.models.simple.Simple3DNet(*args: Any, **kwargs: Any)

Bases: Module

forward(x)

class elektronn3.models.simple.StackedConv2Scalar(*args: Any, **kwargs: Any)

Bases: Module

forward(x)

class elektronn3.models.simple.StackedConv2ScalarWithLatentAdd(*args: Any, **kwargs: Any)

Bases: Module

forward(x, scal)