elektronn3.models.simple module

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

Bases: torch.nn.

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

Bases: torch.nn.

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

Bases: torch.nn.

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

Bases: torch.nn.

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

Bases: torch.nn.

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

Bases: torch.nn.

forward(x, scal)[source]