Identity ======== IDENTITY -------- IdentityModel ~~~~~~~~~~~~~ .. mermaid:: classDiagram class IdentityModel { +int num_spatial_dims +bool register +forward() Tensor } class BaseModel { } <> BaseModel BaseModel <|-- IdentityModel : herits class ModelABC { +bool register +int in_channels +int out_channels +tuple input_shape +check_required_attributes() None } <> ModelABC ModelABC <|-- IdentityModel : herits class Module <> Module Module <|-- IdentityModel : herits .. autosummary:: :toctree: generated :nosignatures: ~mfai.pytorch.models.identity.IdentityModel