lfxai.models.base module

class BlackBox

Bases: Module

_backward_hooks: Dict[int, Callable]

_buffers: Dict[str, Optional[Tensor]]

_forward_hooks: Dict[int, Callable]

_forward_pre_hooks: Dict[int, Callable]

_is_full_backward_hook: Optional[bool]

_load_state_dict_post_hooks: Dict[int, Callable]

_load_state_dict_pre_hooks: Dict[int, Callable]

_modules: Dict[str, Optional[Module]]

_non_persistent_buffers_set: Set[str]

_parameters: Dict[str, Optional[Parameter]]

_state_dict_hooks: Dict[int, Callable]

latent(x: tensor) → tensor

training: bool