Skip to main content
Ctrl+K

AIXD

  • Introduction
  • Installation
  • API Reference
  • User Guide
  • Examples
    • License
    • Authors
    • Citing
    • References
  • Introduction
  • Installation
  • API Reference
  • User Guide
  • Examples
  • License
  • Authors
  • Citing
  • References

Section Navigation

  • AI-eXtended Design
    • aixd.data
      • Dataset
        • Dataset.analysis
        • Dataset.check_data_consistency
        • Dataset.check_dataset_consistency
        • Dataset.data_mat_with_dobjs
        • Dataset.from_dataset_folder
        • Dataset.get_data_objects_by_name
        • Dataset.get_samples
        • Dataset.import_data
        • Dataset.import_data_from_csv
        • Dataset.import_data_from_df
        • Dataset.load
        • Dataset.sampling
        • Dataset.save_dataset_obj
        • Dataset.summary_data
        • Dataset.summary_datablocks
        • Dataset.summary_dataobjects
        • Dataset.update_obj_domains
        • Dataset.write_data_dp_pa
      • DataObject
        • DataObject.append_transformation
        • DataObject.check_data_consistency
        • DataObject.copy
        • DataObject.get_activation_outhead
        • DataObject.get_loss_evaluation
        • DataObject.get_ml_heads
        • DataObject.get_objective
        • DataObject.grid_samples
        • DataObject.has_name
        • DataObject.inverse_transform
        • DataObject.plot_distrib
        • DataObject.prepend_transformation
        • DataObject.print_transf_norm
        • DataObject.random_samples
        • DataObject.sample_around
        • DataObject.transform
        • DataObject.transform_is_fitted
        • DataObject.update_dobj_types
        • DataObject.update_obj
      • DataReal
        • DataReal.from_range
        • DataReal.get_activation_outhead
        • DataReal.get_loss_evaluation
        • DataReal.get_ml_heads
        • DataReal.get_objective
        • DataReal.plot_distrib
        • DataReal.append_transformation
        • DataReal.check_data_consistency
        • DataReal.copy
        • DataReal.grid_samples
        • DataReal.has_name
        • DataReal.inverse_transform
        • DataReal.prepend_transformation
        • DataReal.print_transf_norm
        • DataReal.random_samples
        • DataReal.sample_around
        • DataReal.transform
        • DataReal.transform_is_fitted
        • DataReal.update_dobj_types
        • DataReal.update_obj
      • DataDiscrete
        • DataDiscrete.is_data_transformed
        • DataDiscrete.plot_distrib
        • DataDiscrete.append_transformation
        • DataDiscrete.check_data_consistency
        • DataDiscrete.copy
        • DataDiscrete.get_activation_outhead
        • DataDiscrete.get_loss_evaluation
        • DataDiscrete.get_ml_heads
        • DataDiscrete.get_objective
        • DataDiscrete.grid_samples
        • DataDiscrete.has_name
        • DataDiscrete.inverse_transform
        • DataDiscrete.prepend_transformation
        • DataDiscrete.print_transf_norm
        • DataDiscrete.random_samples
        • DataDiscrete.sample_around
        • DataDiscrete.transform
        • DataDiscrete.transform_is_fitted
        • DataDiscrete.update_dobj_types
        • DataDiscrete.update_obj
      • DataInt
        • DataInt.from_range
        • DataInt.get_activation_outhead
        • DataInt.get_loss_evaluation
        • DataInt.get_ml_heads
        • DataInt.get_objective
        • DataInt.append_transformation
        • DataInt.check_data_consistency
        • DataInt.copy
        • DataInt.grid_samples
        • DataInt.has_name
        • DataInt.inverse_transform
        • DataInt.is_data_transformed
        • DataInt.plot_distrib
        • DataInt.prepend_transformation
        • DataInt.print_transf_norm
        • DataInt.random_samples
        • DataInt.sample_around
        • DataInt.transform
        • DataInt.transform_is_fitted
        • DataInt.update_dobj_types
        • DataInt.update_obj
      • DataCategorical
        • DataCategorical.copy
        • DataCategorical.from_options
        • DataCategorical.get_activation_outhead
        • DataCategorical.get_loss_evaluation
        • DataCategorical.get_ml_heads
        • DataCategorical.get_objective
        • DataCategorical.append_transformation
        • DataCategorical.check_data_consistency
        • DataCategorical.grid_samples
        • DataCategorical.has_name
        • DataCategorical.inverse_transform
        • DataCategorical.is_data_transformed
        • DataCategorical.plot_distrib
        • DataCategorical.prepend_transformation
        • DataCategorical.print_transf_norm
        • DataCategorical.random_samples
        • DataCategorical.sample_around
        • DataCategorical.transform
        • DataCategorical.transform_is_fitted
        • DataCategorical.update_dobj_types
        • DataCategorical.update_obj
      • DataBool
        • DataBool.append_transformation
        • DataBool.check_data_consistency
        • DataBool.copy
        • DataBool.from_options
        • DataBool.get_activation_outhead
        • DataBool.get_loss_evaluation
        • DataBool.get_ml_heads
        • DataBool.get_objective
        • DataBool.grid_samples
        • DataBool.has_name
        • DataBool.inverse_transform
        • DataBool.is_data_transformed
        • DataBool.plot_distrib
        • DataBool.prepend_transformation
        • DataBool.print_transf_norm
        • DataBool.random_samples
        • DataBool.sample_around
        • DataBool.transform
        • DataBool.transform_is_fitted
        • DataBool.update_dobj_types
        • DataBool.update_obj
      • Domain
        • Domain.check_domain_consistency
        • Domain.copy
        • Domain.domain_to_normalised
        • Domain.grid_samples
        • Domain.normalised_to_domain
        • Domain.random_samples
        • Domain.sample_around
        • Domain.update_domain
      • Options
        • Options.check_domain_consistency
        • Options.copy
        • Options.domain_to_normalised
        • Options.grid_samples
        • Options.normalised_to_domain
        • Options.random_samples
        • Options.sample_around
        • Options.update_domain
      • Interval
        • Interval.check_domain_consistency
        • Interval.copy
        • Interval.domain_to_normalised
        • Interval.grid_samples
        • Interval.normalised_to_domain
        • Interval.random_samples
        • Interval.sample_around
        • Interval.update_domain
      • IntervalMasked
        • IntervalMasked.copy
        • IntervalMasked.check_domain_consistency
        • IntervalMasked.domain_to_normalised
        • IntervalMasked.grid_samples
        • IntervalMasked.normalised_to_domain
        • IntervalMasked.random_samples
        • IntervalMasked.sample_around
        • IntervalMasked.update_domain
      • DataBlock
        • DataBlock.check_data_consistency
        • DataBlock.get_cols_dobjs
        • DataBlock.get_data_mat
        • DataBlock.get_data_mat_dtypes
        • DataBlock.get_data_mats
        • DataBlock.get_dobjs
        • DataBlock.info_dobj
        • DataBlock.print_objects
        • DataBlock.print_transf
        • DataBlock.read
        • DataBlock.update_obj
        • DataBlock.update_position_indices
        • DataBlock.write
      • DesignParameters
        • DesignParameters.check_data_consistency
        • DesignParameters.get_cols_dobjs
        • DesignParameters.get_data_mat
        • DesignParameters.get_data_mat_dtypes
        • DesignParameters.get_data_mats
        • DesignParameters.get_dobjs
        • DesignParameters.info_dobj
        • DesignParameters.print_objects
        • DesignParameters.print_transf
        • DesignParameters.read
        • DesignParameters.update_obj
        • DesignParameters.update_position_indices
        • DesignParameters.write
      • PerformanceAttributes
        • PerformanceAttributes.check_data_consistency
        • PerformanceAttributes.get_cols_dobjs
        • PerformanceAttributes.get_data_mat
        • PerformanceAttributes.get_data_mat_dtypes
        • PerformanceAttributes.get_data_mats
        • PerformanceAttributes.get_dobjs
        • PerformanceAttributes.info_dobj
        • PerformanceAttributes.print_objects
        • PerformanceAttributes.print_transf
        • PerformanceAttributes.read
        • PerformanceAttributes.update_obj
        • PerformanceAttributes.update_position_indices
        • PerformanceAttributes.write
      • DesignRepresentation
        • DesignRepresentation.read
        • DesignRepresentation.write
        • DesignRepresentation.check_data_consistency
        • DesignRepresentation.get_cols_dobjs
        • DesignRepresentation.get_data_mat
        • DesignRepresentation.get_data_mat_dtypes
        • DesignRepresentation.get_data_mats
        • DesignRepresentation.get_dobjs
        • DesignRepresentation.info_dobj
        • DesignRepresentation.print_objects
        • DesignRepresentation.print_transf
        • DesignRepresentation.update_obj
        • DesignRepresentation.update_position_indices
      • TransformableDataBlock
        • TransformableDataBlock.get_dobj_dimensions
        • TransformableDataBlock.inverse_transform
        • TransformableDataBlock.to_data_block
        • TransformableDataBlock.transform
        • TransformableDataBlock.transformation_is_fitted
        • TransformableDataBlock.check_data_consistency
        • TransformableDataBlock.get_cols_dobjs
        • TransformableDataBlock.get_data_mat
        • TransformableDataBlock.get_data_mat_dtypes
        • TransformableDataBlock.get_data_mats
        • TransformableDataBlock.get_dobjs
        • TransformableDataBlock.info_dobj
        • TransformableDataBlock.print_objects
        • TransformableDataBlock.print_transf
        • TransformableDataBlock.read
        • TransformableDataBlock.update_obj
        • TransformableDataBlock.update_position_indices
        • TransformableDataBlock.write
      • InputML
        • InputML.check_data_consistency
        • InputML.get_cols_dobjs
        • InputML.get_data_mat
        • InputML.get_data_mat_dtypes
        • InputML.get_data_mats
        • InputML.get_dobj_dimensions
        • InputML.get_dobjs
        • InputML.info_dobj
        • InputML.inverse_transform
        • InputML.print_objects
        • InputML.print_transf
        • InputML.read
        • InputML.to_data_block
        • InputML.transform
        • InputML.transformation_is_fitted
        • InputML.update_obj
        • InputML.update_position_indices
        • InputML.write
      • OutputML
        • OutputML.check_data_consistency
        • OutputML.get_cols_dobjs
        • OutputML.get_data_mat
        • OutputML.get_data_mat_dtypes
        • OutputML.get_data_mats
        • OutputML.get_dobj_dimensions
        • OutputML.get_dobjs
        • OutputML.info_dobj
        • OutputML.inverse_transform
        • OutputML.print_objects
        • OutputML.print_transf
        • OutputML.read
        • OutputML.to_data_block
        • OutputML.transform
        • OutputML.transformation_is_fitted
        • OutputML.update_obj
        • OutputML.update_position_indices
        • OutputML.write
      • CustomCallback
        • CustomCallback.run
        • CustomCallback.to_numpy
      • AnalysisCallback
        • AnalysisCallback.analyze
        • AnalysisCallback.run
        • AnalysisCallback.to_numpy
      • DataloaderCallback
        • DataloaderCallback.run
        • DataloaderCallback.to_numpy
      • ImportCallback
        • ImportCallback.run
        • ImportCallback.to_numpy
      • PostGenerationCallback
        • PostGenerationCallback.run
        • PostGenerationCallback.to_numpy
      • SamplingCallback
        • SamplingCallback.run
        • SamplingCallback.to_numpy
      • DataObjectTransform
        • DataObjectTransform.copy
        • DataObjectTransform.fit
        • DataObjectTransform.fit_transform
        • DataObjectTransform.inverse_transform
        • DataObjectTransform.is_fitted
        • DataObjectTransform.reset
        • DataObjectTransform.transform
      • Log10Transform
        • Log10Transform.inverse_transform
        • Log10Transform.transform
        • Log10Transform.copy
        • Log10Transform.fit
        • Log10Transform.fit_transform
        • Log10Transform.is_fitted
        • Log10Transform.reset
      • SigmoidTransform
        • SigmoidTransform.inverse_transform
        • SigmoidTransform.is_fitted
        • SigmoidTransform.reset
        • SigmoidTransform.transform
        • SigmoidTransform.copy
        • SigmoidTransform.fit
        • SigmoidTransform.fit_transform
      • Standardization
        • Standardization.inverse_transform
        • Standardization.is_fitted
        • Standardization.reset
        • Standardization.transform
        • Standardization.copy
        • Standardization.fit
        • Standardization.fit_transform
      • MinMaxScaler
        • MinMaxScaler.inverse_transform
        • MinMaxScaler.is_fitted
        • MinMaxScaler.reset
        • MinMaxScaler.transform
        • MinMaxScaler.copy
        • MinMaxScaler.fit
        • MinMaxScaler.fit_transform
      • ZeroToOne
        • ZeroToOne.copy
        • ZeroToOne.fit
        • ZeroToOne.fit_transform
        • ZeroToOne.inverse_transform
        • ZeroToOne.is_fitted
        • ZeroToOne.reset
        • ZeroToOne.transform
      • MinusOneToOne
        • MinusOneToOne.copy
        • MinusOneToOne.fit
        • MinusOneToOne.fit_transform
        • MinusOneToOne.inverse_transform
        • MinusOneToOne.is_fitted
        • MinusOneToOne.reset
        • MinusOneToOne.transform
      • MaskedMinMaxScaler
        • MaskedMinMaxScaler.inverse_transform
        • MaskedMinMaxScaler.transform
        • MaskedMinMaxScaler.copy
        • MaskedMinMaxScaler.fit
        • MaskedMinMaxScaler.fit_transform
        • MaskedMinMaxScaler.is_fitted
        • MaskedMinMaxScaler.reset
      • MaskedZeroToOne
        • MaskedZeroToOne.copy
        • MaskedZeroToOne.fit
        • MaskedZeroToOne.fit_transform
        • MaskedZeroToOne.inverse_transform
        • MaskedZeroToOne.is_fitted
        • MaskedZeroToOne.reset
        • MaskedZeroToOne.transform
      • MaskedMinusOneToOne
        • MaskedMinusOneToOne.copy
        • MaskedMinusOneToOne.fit
        • MaskedMinusOneToOne.fit_transform
        • MaskedMinusOneToOne.inverse_transform
        • MaskedMinusOneToOne.is_fitted
        • MaskedMinusOneToOne.reset
        • MaskedMinusOneToOne.transform
      • LabelEncoder
        • LabelEncoder.fit
        • LabelEncoder.inverse_transform
        • LabelEncoder.is_fitted
        • LabelEncoder.reset
        • LabelEncoder.transform
        • LabelEncoder.copy
        • LabelEncoder.fit_transform
      • ToFloat
        • ToFloat.inverse_transform
        • ToFloat.is_fitted
        • ToFloat.transform
        • ToFloat.copy
        • ToFloat.fit
        • ToFloat.fit_transform
        • ToFloat.reset
      • aixd.data.utils_data.convert_to
      • aixd.data.utils_data.combine_formats
      • aixd.data.utils_data.reformat_dict_to_dataframe
      • aixd.data.utils_data.reformat_dataframe_to_dict
      • aixd.data.utils_data.reformat_dataframe_to_dictlist
      • aixd.data.utils_data.reformat_dataframeflat_to_dict
      • aixd.data.utils_data.reformat_dataframe_to_dataframeflat
      • aixd.data.utils_data.reformat_dataframeflat_to_dataframe
      • aixd.data.utils_data.reformat_list_to_array
      • aixd.data.utils_data.reformat_array_to_list
      • aixd.data.utils_data.reformat_list_to_dataframe
      • aixd.data.utils_data.reformat_dataframe_to_list
      • aixd.data.utils_data.reformat_list_to_dict
      • aixd.data.utils_data.reformat_list_to_dictlist
      • aixd.data.utils_data.reformat_dictlist_to_dict
      • aixd.data.utils_data.reformat_dict_to_dictlist
      • aixd.data.utils_data.reformat_array_to_torch
      • aixd.data.utils_data.reformat_torch_to_array
    • aixd.mlmodel
      • DataModule
        • DataModule.convert_to_dataloader
        • DataModule.from_dataset
        • DataModule.from_parameters
        • DataModule.get_checksum
        • DataModule.get_parameters
        • DataModule.inverse_transform
        • DataModule.inverse_transform_x
        • DataModule.inverse_transform_y
        • DataModule.predict_dataloader
        • DataModule.summary_input_output_dimensions
        • DataModule.test_dataloader
        • DataModule.train_dataloader
        • DataModule.transform
        • DataModule.transform_x
        • DataModule.transform_y
        • DataModule.val_dataloader
      • CondAEModel
        • CondAEModel.calc_x_losses_calculation
        • CondAEModel.calc_y_losses_calculation
        • CondAEModel.configure_optimizers
        • CondAEModel.decode
        • CondAEModel.encode
        • CondAEModel.evaluate
        • CondAEModel.fit
        • CondAEModel.forward
        • CondAEModel.forward_evaluation
        • CondAEModel.from_datamodule
        • CondAEModel.generate
        • CondAEModel.get_evaluation_losses
        • CondAEModel.load_model_from_checkpoint
        • CondAEModel.on_load_checkpoint
        • CondAEModel.on_save_checkpoint
        • CondAEModel.predict
        • CondAEModel.predict_step
        • CondAEModel.predict_y
        • CondAEModel.save_extra_parameters
        • CondAEModel.split_x_head_in
        • CondAEModel.split_x_head_out
        • CondAEModel.split_xy_in
        • CondAEModel.split_xy_out
        • CondAEModel.split_y_head_in
        • CondAEModel.split_y_head_out
        • CondAEModel.summary
        • CondAEModel.test
        • CondAEModel.test_step
        • CondAEModel.to_untransformed_pred
        • CondAEModel.training_step
        • CondAEModel.validate
        • CondAEModel.validation_step
      • CondVAEModel
        • CondVAEModel.calc_x_losses_calculation
        • CondVAEModel.calc_y_losses_calculation
        • CondVAEModel.configure_optimizers
        • CondVAEModel.decode
        • CondVAEModel.encode
        • CondVAEModel.evaluate
        • CondVAEModel.fit
        • CondVAEModel.forward
        • CondVAEModel.forward_evaluation
        • CondVAEModel.from_datamodule
        • CondVAEModel.generate
        • CondVAEModel.get_evaluation_losses
        • CondVAEModel.load_model_from_checkpoint
        • CondVAEModel.on_load_checkpoint
        • CondVAEModel.on_save_checkpoint
        • CondVAEModel.predict
        • CondVAEModel.predict_step
        • CondVAEModel.predict_y
        • CondVAEModel.save_extra_parameters
        • CondVAEModel.split_x_head_in
        • CondVAEModel.split_x_head_out
        • CondVAEModel.split_xy_in
        • CondVAEModel.split_xy_out
        • CondVAEModel.split_y_head_in
        • CondVAEModel.split_y_head_out
        • CondVAEModel.summary
        • CondVAEModel.test
        • CondVAEModel.test_step
        • CondVAEModel.to_untransformed_pred
        • CondVAEModel.training_step
        • CondVAEModel.validate
        • CondVAEModel.validation_step
      • InverseModel
        • InverseModel.fit
        • InverseModel.forward
      • FreezeEncoder
        • FreezeEncoder.finetune_function
        • FreezeEncoder.freeze_before_training
      • ResBlockFC
        • ResBlockFC.forward
        • ResBlockFC.add_module
        • ResBlockFC.apply
        • ResBlockFC.bfloat16
        • ResBlockFC.buffers
        • ResBlockFC.children
        • ResBlockFC.compile
        • ResBlockFC.cpu
        • ResBlockFC.cuda
        • ResBlockFC.double
        • ResBlockFC.eval
        • ResBlockFC.extra_repr
        • ResBlockFC.float
        • ResBlockFC.get_buffer
        • ResBlockFC.get_extra_state
        • ResBlockFC.get_parameter
        • ResBlockFC.get_submodule
        • ResBlockFC.half
        • ResBlockFC.ipu
        • ResBlockFC.load_state_dict
        • ResBlockFC.modules
        • ResBlockFC.mtia
        • ResBlockFC.named_buffers
        • ResBlockFC.named_children
        • ResBlockFC.named_modules
        • ResBlockFC.named_parameters
        • ResBlockFC.parameters
        • ResBlockFC.register_backward_hook
        • ResBlockFC.register_buffer
        • ResBlockFC.register_forward_hook
        • ResBlockFC.register_forward_pre_hook
        • ResBlockFC.register_full_backward_hook
        • ResBlockFC.register_full_backward_pre_hook
        • ResBlockFC.register_load_state_dict_post_hook
        • ResBlockFC.register_load_state_dict_pre_hook
        • ResBlockFC.register_module
        • ResBlockFC.register_parameter
        • ResBlockFC.register_state_dict_post_hook
        • ResBlockFC.register_state_dict_pre_hook
        • ResBlockFC.requires_grad_
        • ResBlockFC.set_extra_state
        • ResBlockFC.set_submodule
        • ResBlockFC.share_memory
        • ResBlockFC.state_dict
        • ResBlockFC.to
        • ResBlockFC.to_empty
        • ResBlockFC.train
        • ResBlockFC.type
        • ResBlockFC.xpu
        • ResBlockFC.zero_grad
      • ResBlockConv
        • ResBlockConv.forward
        • ResBlockConv.add_module
        • ResBlockConv.apply
        • ResBlockConv.bfloat16
        • ResBlockConv.buffers
        • ResBlockConv.children
        • ResBlockConv.compile
        • ResBlockConv.cpu
        • ResBlockConv.cuda
        • ResBlockConv.double
        • ResBlockConv.eval
        • ResBlockConv.extra_repr
        • ResBlockConv.float
        • ResBlockConv.get_buffer
        • ResBlockConv.get_extra_state
        • ResBlockConv.get_parameter
        • ResBlockConv.get_submodule
        • ResBlockConv.half
        • ResBlockConv.ipu
        • ResBlockConv.load_state_dict
        • ResBlockConv.modules
        • ResBlockConv.mtia
        • ResBlockConv.named_buffers
        • ResBlockConv.named_children
        • ResBlockConv.named_modules
        • ResBlockConv.named_parameters
        • ResBlockConv.parameters
        • ResBlockConv.register_backward_hook
        • ResBlockConv.register_buffer
        • ResBlockConv.register_forward_hook
        • ResBlockConv.register_forward_pre_hook
        • ResBlockConv.register_full_backward_hook
        • ResBlockConv.register_full_backward_pre_hook
        • ResBlockConv.register_load_state_dict_post_hook
        • ResBlockConv.register_load_state_dict_pre_hook
        • ResBlockConv.register_module
        • ResBlockConv.register_parameter
        • ResBlockConv.register_state_dict_post_hook
        • ResBlockConv.register_state_dict_pre_hook
        • ResBlockConv.requires_grad_
        • ResBlockConv.set_extra_state
        • ResBlockConv.set_submodule
        • ResBlockConv.share_memory
        • ResBlockConv.state_dict
        • ResBlockConv.to
        • ResBlockConv.to_empty
        • ResBlockConv.train
        • ResBlockConv.type
        • ResBlockConv.xpu
        • ResBlockConv.zero_grad
      • ResBlock1D
        • ResBlock1D.add_module
        • ResBlock1D.apply
        • ResBlock1D.bfloat16
        • ResBlock1D.buffers
        • ResBlock1D.children
        • ResBlock1D.compile
        • ResBlock1D.cpu
        • ResBlock1D.cuda
        • ResBlock1D.double
        • ResBlock1D.eval
        • ResBlock1D.extra_repr
        • ResBlock1D.float
        • ResBlock1D.forward
        • ResBlock1D.get_buffer
        • ResBlock1D.get_extra_state
        • ResBlock1D.get_parameter
        • ResBlock1D.get_submodule
        • ResBlock1D.half
        • ResBlock1D.ipu
        • ResBlock1D.load_state_dict
        • ResBlock1D.modules
        • ResBlock1D.mtia
        • ResBlock1D.named_buffers
        • ResBlock1D.named_children
        • ResBlock1D.named_modules
        • ResBlock1D.named_parameters
        • ResBlock1D.parameters
        • ResBlock1D.register_backward_hook
        • ResBlock1D.register_buffer
        • ResBlock1D.register_forward_hook
        • ResBlock1D.register_forward_pre_hook
        • ResBlock1D.register_full_backward_hook
        • ResBlock1D.register_full_backward_pre_hook
        • ResBlock1D.register_load_state_dict_post_hook
        • ResBlock1D.register_load_state_dict_pre_hook
        • ResBlock1D.register_module
        • ResBlock1D.register_parameter
        • ResBlock1D.register_state_dict_post_hook
        • ResBlock1D.register_state_dict_pre_hook
        • ResBlock1D.requires_grad_
        • ResBlock1D.set_extra_state
        • ResBlock1D.set_submodule
        • ResBlock1D.share_memory
        • ResBlock1D.state_dict
        • ResBlock1D.to
        • ResBlock1D.to_empty
        • ResBlock1D.train
        • ResBlock1D.type
        • ResBlock1D.xpu
        • ResBlock1D.zero_grad
      • ResBlock2D
        • ResBlock2D.add_module
        • ResBlock2D.apply
        • ResBlock2D.bfloat16
        • ResBlock2D.buffers
        • ResBlock2D.children
        • ResBlock2D.compile
        • ResBlock2D.cpu
        • ResBlock2D.cuda
        • ResBlock2D.double
        • ResBlock2D.eval
        • ResBlock2D.extra_repr
        • ResBlock2D.float
        • ResBlock2D.forward
        • ResBlock2D.get_buffer
        • ResBlock2D.get_extra_state
        • ResBlock2D.get_parameter
        • ResBlock2D.get_submodule
        • ResBlock2D.half
        • ResBlock2D.ipu
        • ResBlock2D.load_state_dict
        • ResBlock2D.modules
        • ResBlock2D.mtia
        • ResBlock2D.named_buffers
        • ResBlock2D.named_children
        • ResBlock2D.named_modules
        • ResBlock2D.named_parameters
        • ResBlock2D.parameters
        • ResBlock2D.register_backward_hook
        • ResBlock2D.register_buffer
        • ResBlock2D.register_forward_hook
        • ResBlock2D.register_forward_pre_hook
        • ResBlock2D.register_full_backward_hook
        • ResBlock2D.register_full_backward_pre_hook
        • ResBlock2D.register_load_state_dict_post_hook
        • ResBlock2D.register_load_state_dict_pre_hook
        • ResBlock2D.register_module
        • ResBlock2D.register_parameter
        • ResBlock2D.register_state_dict_post_hook
        • ResBlock2D.register_state_dict_pre_hook
        • ResBlock2D.requires_grad_
        • ResBlock2D.set_extra_state
        • ResBlock2D.set_submodule
        • ResBlock2D.share_memory
        • ResBlock2D.state_dict
        • ResBlock2D.to
        • ResBlock2D.to_empty
        • ResBlock2D.train
        • ResBlock2D.type
        • ResBlock2D.xpu
        • ResBlock2D.zero_grad
      • ResBlock3D
        • ResBlock3D.add_module
        • ResBlock3D.apply
        • ResBlock3D.bfloat16
        • ResBlock3D.buffers
        • ResBlock3D.children
        • ResBlock3D.compile
        • ResBlock3D.cpu
        • ResBlock3D.cuda
        • ResBlock3D.double
        • ResBlock3D.eval
        • ResBlock3D.extra_repr
        • ResBlock3D.float
        • ResBlock3D.forward
        • ResBlock3D.get_buffer
        • ResBlock3D.get_extra_state
        • ResBlock3D.get_parameter
        • ResBlock3D.get_submodule
        • ResBlock3D.half
        • ResBlock3D.ipu
        • ResBlock3D.load_state_dict
        • ResBlock3D.modules
        • ResBlock3D.mtia
        • ResBlock3D.named_buffers
        • ResBlock3D.named_children
        • ResBlock3D.named_modules
        • ResBlock3D.named_parameters
        • ResBlock3D.parameters
        • ResBlock3D.register_backward_hook
        • ResBlock3D.register_buffer
        • ResBlock3D.register_forward_hook
        • ResBlock3D.register_forward_pre_hook
        • ResBlock3D.register_full_backward_hook
        • ResBlock3D.register_full_backward_pre_hook
        • ResBlock3D.register_load_state_dict_post_hook
        • ResBlock3D.register_load_state_dict_pre_hook
        • ResBlock3D.register_module
        • ResBlock3D.register_parameter
        • ResBlock3D.register_state_dict_post_hook
        • ResBlock3D.register_state_dict_pre_hook
        • ResBlock3D.requires_grad_
        • ResBlock3D.set_extra_state
        • ResBlock3D.set_submodule
        • ResBlock3D.share_memory
        • ResBlock3D.state_dict
        • ResBlock3D.to
        • ResBlock3D.to_empty
        • ResBlock3D.train
        • ResBlock3D.type
        • ResBlock3D.xpu
        • ResBlock3D.zero_grad
      • SelfAttn
        • SelfAttn.forward
        • SelfAttn.add_module
        • SelfAttn.apply
        • SelfAttn.bfloat16
        • SelfAttn.buffers
        • SelfAttn.children
        • SelfAttn.compile
        • SelfAttn.cpu
        • SelfAttn.cuda
        • SelfAttn.double
        • SelfAttn.eval
        • SelfAttn.extra_repr
        • SelfAttn.float
        • SelfAttn.get_buffer
        • SelfAttn.get_extra_state
        • SelfAttn.get_parameter
        • SelfAttn.get_submodule
        • SelfAttn.half
        • SelfAttn.ipu
        • SelfAttn.load_state_dict
        • SelfAttn.modules
        • SelfAttn.mtia
        • SelfAttn.named_buffers
        • SelfAttn.named_children
        • SelfAttn.named_modules
        • SelfAttn.named_parameters
        • SelfAttn.parameters
        • SelfAttn.register_backward_hook
        • SelfAttn.register_buffer
        • SelfAttn.register_forward_hook
        • SelfAttn.register_forward_pre_hook
        • SelfAttn.register_full_backward_hook
        • SelfAttn.register_full_backward_pre_hook
        • SelfAttn.register_load_state_dict_post_hook
        • SelfAttn.register_load_state_dict_pre_hook
        • SelfAttn.register_module
        • SelfAttn.register_parameter
        • SelfAttn.register_state_dict_post_hook
        • SelfAttn.register_state_dict_pre_hook
        • SelfAttn.requires_grad_
        • SelfAttn.set_extra_state
        • SelfAttn.set_submodule
        • SelfAttn.share_memory
        • SelfAttn.state_dict
        • SelfAttn.to
        • SelfAttn.to_empty
        • SelfAttn.train
        • SelfAttn.type
        • SelfAttn.xpu
        • SelfAttn.zero_grad
      • SelfAttn1D
        • SelfAttn1D.add_module
        • SelfAttn1D.apply
        • SelfAttn1D.bfloat16
        • SelfAttn1D.buffers
        • SelfAttn1D.children
        • SelfAttn1D.compile
        • SelfAttn1D.cpu
        • SelfAttn1D.cuda
        • SelfAttn1D.double
        • SelfAttn1D.eval
        • SelfAttn1D.extra_repr
        • SelfAttn1D.float
        • SelfAttn1D.forward
        • SelfAttn1D.get_buffer
        • SelfAttn1D.get_extra_state
        • SelfAttn1D.get_parameter
        • SelfAttn1D.get_submodule
        • SelfAttn1D.half
        • SelfAttn1D.ipu
        • SelfAttn1D.load_state_dict
        • SelfAttn1D.modules
        • SelfAttn1D.mtia
        • SelfAttn1D.named_buffers
        • SelfAttn1D.named_children
        • SelfAttn1D.named_modules
        • SelfAttn1D.named_parameters
        • SelfAttn1D.parameters
        • SelfAttn1D.register_backward_hook
        • SelfAttn1D.register_buffer
        • SelfAttn1D.register_forward_hook
        • SelfAttn1D.register_forward_pre_hook
        • SelfAttn1D.register_full_backward_hook
        • SelfAttn1D.register_full_backward_pre_hook
        • SelfAttn1D.register_load_state_dict_post_hook
        • SelfAttn1D.register_load_state_dict_pre_hook
        • SelfAttn1D.register_module
        • SelfAttn1D.register_parameter
        • SelfAttn1D.register_state_dict_post_hook
        • SelfAttn1D.register_state_dict_pre_hook
        • SelfAttn1D.requires_grad_
        • SelfAttn1D.set_extra_state
        • SelfAttn1D.set_submodule
        • SelfAttn1D.share_memory
        • SelfAttn1D.state_dict
        • SelfAttn1D.to
        • SelfAttn1D.to_empty
        • SelfAttn1D.train
        • SelfAttn1D.type
        • SelfAttn1D.xpu
        • SelfAttn1D.zero_grad
      • SelfAttn2D
        • SelfAttn2D.add_module
        • SelfAttn2D.apply
        • SelfAttn2D.bfloat16
        • SelfAttn2D.buffers
        • SelfAttn2D.children
        • SelfAttn2D.compile
        • SelfAttn2D.cpu
        • SelfAttn2D.cuda
        • SelfAttn2D.double
        • SelfAttn2D.eval
        • SelfAttn2D.extra_repr
        • SelfAttn2D.float
        • SelfAttn2D.forward
        • SelfAttn2D.get_buffer
        • SelfAttn2D.get_extra_state
        • SelfAttn2D.get_parameter
        • SelfAttn2D.get_submodule
        • SelfAttn2D.half
        • SelfAttn2D.ipu
        • SelfAttn2D.load_state_dict
        • SelfAttn2D.modules
        • SelfAttn2D.mtia
        • SelfAttn2D.named_buffers
        • SelfAttn2D.named_children
        • SelfAttn2D.named_modules
        • SelfAttn2D.named_parameters
        • SelfAttn2D.parameters
        • SelfAttn2D.register_backward_hook
        • SelfAttn2D.register_buffer
        • SelfAttn2D.register_forward_hook
        • SelfAttn2D.register_forward_pre_hook
        • SelfAttn2D.register_full_backward_hook
        • SelfAttn2D.register_full_backward_pre_hook
        • SelfAttn2D.register_load_state_dict_post_hook
        • SelfAttn2D.register_load_state_dict_pre_hook
        • SelfAttn2D.register_module
        • SelfAttn2D.register_parameter
        • SelfAttn2D.register_state_dict_post_hook
        • SelfAttn2D.register_state_dict_pre_hook
        • SelfAttn2D.requires_grad_
        • SelfAttn2D.set_extra_state
        • SelfAttn2D.set_submodule
        • SelfAttn2D.share_memory
        • SelfAttn2D.state_dict
        • SelfAttn2D.to
        • SelfAttn2D.to_empty
        • SelfAttn2D.train
        • SelfAttn2D.type
        • SelfAttn2D.xpu
        • SelfAttn2D.zero_grad
      • SelfAttn3D
        • SelfAttn3D.add_module
        • SelfAttn3D.apply
        • SelfAttn3D.bfloat16
        • SelfAttn3D.buffers
        • SelfAttn3D.children
        • SelfAttn3D.compile
        • SelfAttn3D.cpu
        • SelfAttn3D.cuda
        • SelfAttn3D.double
        • SelfAttn3D.eval
        • SelfAttn3D.extra_repr
        • SelfAttn3D.float
        • SelfAttn3D.forward
        • SelfAttn3D.get_buffer
        • SelfAttn3D.get_extra_state
        • SelfAttn3D.get_parameter
        • SelfAttn3D.get_submodule
        • SelfAttn3D.half
        • SelfAttn3D.ipu
        • SelfAttn3D.load_state_dict
        • SelfAttn3D.modules
        • SelfAttn3D.mtia
        • SelfAttn3D.named_buffers
        • SelfAttn3D.named_children
        • SelfAttn3D.named_modules
        • SelfAttn3D.named_parameters
        • SelfAttn3D.parameters
        • SelfAttn3D.register_backward_hook
        • SelfAttn3D.register_buffer
        • SelfAttn3D.register_forward_hook
        • SelfAttn3D.register_forward_pre_hook
        • SelfAttn3D.register_full_backward_hook
        • SelfAttn3D.register_full_backward_pre_hook
        • SelfAttn3D.register_load_state_dict_post_hook
        • SelfAttn3D.register_load_state_dict_pre_hook
        • SelfAttn3D.register_module
        • SelfAttn3D.register_parameter
        • SelfAttn3D.register_state_dict_post_hook
        • SelfAttn3D.register_state_dict_pre_hook
        • SelfAttn3D.requires_grad_
        • SelfAttn3D.set_extra_state
        • SelfAttn3D.set_submodule
        • SelfAttn3D.share_memory
        • SelfAttn3D.state_dict
        • SelfAttn3D.to
        • SelfAttn3D.to_empty
        • SelfAttn3D.train
        • SelfAttn3D.type
        • SelfAttn3D.xpu
        • SelfAttn3D.zero_grad
      • Activation
        • Activation.forward
        • Activation.add_module
        • Activation.apply
        • Activation.bfloat16
        • Activation.buffers
        • Activation.children
        • Activation.compile
        • Activation.cpu
        • Activation.cuda
        • Activation.double
        • Activation.eval
        • Activation.extra_repr
        • Activation.float
        • Activation.get_buffer
        • Activation.get_extra_state
        • Activation.get_parameter
        • Activation.get_submodule
        • Activation.half
        • Activation.ipu
        • Activation.load_state_dict
        • Activation.modules
        • Activation.mtia
        • Activation.named_buffers
        • Activation.named_children
        • Activation.named_modules
        • Activation.named_parameters
        • Activation.parameters
        • Activation.register_backward_hook
        • Activation.register_buffer
        • Activation.register_forward_hook
        • Activation.register_forward_pre_hook
        • Activation.register_full_backward_hook
        • Activation.register_full_backward_pre_hook
        • Activation.register_load_state_dict_post_hook
        • Activation.register_load_state_dict_pre_hook
        • Activation.register_module
        • Activation.register_parameter
        • Activation.register_state_dict_post_hook
        • Activation.register_state_dict_pre_hook
        • Activation.requires_grad_
        • Activation.set_extra_state
        • Activation.set_submodule
        • Activation.share_memory
        • Activation.state_dict
        • Activation.to
        • Activation.to_empty
        • Activation.train
        • Activation.type
        • Activation.xpu
        • Activation.zero_grad
      • Decoder
        • Decoder.forward
        • Decoder.summary
        • Decoder.add_module
        • Decoder.apply
        • Decoder.bfloat16
        • Decoder.buffers
        • Decoder.children
        • Decoder.compile
        • Decoder.cpu
        • Decoder.cuda
        • Decoder.double
        • Decoder.eval
        • Decoder.extra_repr
        • Decoder.float
        • Decoder.get_buffer
        • Decoder.get_extra_state
        • Decoder.get_parameter
        • Decoder.get_submodule
        • Decoder.half
        • Decoder.ipu
        • Decoder.load_state_dict
        • Decoder.modules
        • Decoder.mtia
        • Decoder.named_buffers
        • Decoder.named_children
        • Decoder.named_modules
        • Decoder.named_parameters
        • Decoder.parameters
        • Decoder.register_backward_hook
        • Decoder.register_buffer
        • Decoder.register_forward_hook
        • Decoder.register_forward_pre_hook
        • Decoder.register_full_backward_hook
        • Decoder.register_full_backward_pre_hook
        • Decoder.register_load_state_dict_post_hook
        • Decoder.register_load_state_dict_pre_hook
        • Decoder.register_module
        • Decoder.register_parameter
        • Decoder.register_state_dict_post_hook
        • Decoder.register_state_dict_pre_hook
        • Decoder.requires_grad_
        • Decoder.set_extra_state
        • Decoder.set_submodule
        • Decoder.share_memory
        • Decoder.state_dict
        • Decoder.to
        • Decoder.to_empty
        • Decoder.train
        • Decoder.type
        • Decoder.xpu
        • Decoder.zero_grad
      • Encoder
        • Encoder.forward
        • Encoder.summary
        • Encoder.add_module
        • Encoder.apply
        • Encoder.bfloat16
        • Encoder.buffers
        • Encoder.children
        • Encoder.compile
        • Encoder.cpu
        • Encoder.cuda
        • Encoder.double
        • Encoder.eval
        • Encoder.extra_repr
        • Encoder.float
        • Encoder.get_buffer
        • Encoder.get_extra_state
        • Encoder.get_parameter
        • Encoder.get_submodule
        • Encoder.half
        • Encoder.ipu
        • Encoder.load_state_dict
        • Encoder.modules
        • Encoder.mtia
        • Encoder.named_buffers
        • Encoder.named_children
        • Encoder.named_modules
        • Encoder.named_parameters
        • Encoder.parameters
        • Encoder.register_backward_hook
        • Encoder.register_buffer
        • Encoder.register_forward_hook
        • Encoder.register_forward_pre_hook
        • Encoder.register_full_backward_hook
        • Encoder.register_full_backward_pre_hook
        • Encoder.register_load_state_dict_post_hook
        • Encoder.register_load_state_dict_pre_hook
        • Encoder.register_module
        • Encoder.register_parameter
        • Encoder.register_state_dict_post_hook
        • Encoder.register_state_dict_pre_hook
        • Encoder.requires_grad_
        • Encoder.set_extra_state
        • Encoder.set_submodule
        • Encoder.share_memory
        • Encoder.state_dict
        • Encoder.to
        • Encoder.to_empty
        • Encoder.train
        • Encoder.type
        • Encoder.xpu
        • Encoder.zero_grad
      • VEncoder
        • VEncoder.forward
        • VEncoder.add_module
        • VEncoder.apply
        • VEncoder.bfloat16
        • VEncoder.buffers
        • VEncoder.children
        • VEncoder.compile
        • VEncoder.cpu
        • VEncoder.cuda
        • VEncoder.double
        • VEncoder.eval
        • VEncoder.extra_repr
        • VEncoder.float
        • VEncoder.get_buffer
        • VEncoder.get_extra_state
        • VEncoder.get_parameter
        • VEncoder.get_submodule
        • VEncoder.half
        • VEncoder.ipu
        • VEncoder.load_state_dict
        • VEncoder.modules
        • VEncoder.mtia
        • VEncoder.named_buffers
        • VEncoder.named_children
        • VEncoder.named_modules
        • VEncoder.named_parameters
        • VEncoder.parameters
        • VEncoder.register_backward_hook
        • VEncoder.register_buffer
        • VEncoder.register_forward_hook
        • VEncoder.register_forward_pre_hook
        • VEncoder.register_full_backward_hook
        • VEncoder.register_full_backward_pre_hook
        • VEncoder.register_load_state_dict_post_hook
        • VEncoder.register_load_state_dict_pre_hook
        • VEncoder.register_module
        • VEncoder.register_parameter
        • VEncoder.register_state_dict_post_hook
        • VEncoder.register_state_dict_pre_hook
        • VEncoder.requires_grad_
        • VEncoder.set_extra_state
        • VEncoder.set_submodule
        • VEncoder.share_memory
        • VEncoder.state_dict
        • VEncoder.summary
        • VEncoder.to
        • VEncoder.to_empty
        • VEncoder.train
        • VEncoder.type
        • VEncoder.xpu
        • VEncoder.zero_grad
      • InHeadFC
        • InHeadFC.forward
        • InHeadFC.add_module
        • InHeadFC.apply
        • InHeadFC.bfloat16
        • InHeadFC.buffers
        • InHeadFC.children
        • InHeadFC.compile
        • InHeadFC.cpu
        • InHeadFC.cuda
        • InHeadFC.double
        • InHeadFC.eval
        • InHeadFC.extra_repr
        • InHeadFC.float
        • InHeadFC.get_buffer
        • InHeadFC.get_extra_state
        • InHeadFC.get_parameter
        • InHeadFC.get_submodule
        • InHeadFC.half
        • InHeadFC.ipu
        • InHeadFC.load_state_dict
        • InHeadFC.modules
        • InHeadFC.mtia
        • InHeadFC.named_buffers
        • InHeadFC.named_children
        • InHeadFC.named_modules
        • InHeadFC.named_parameters
        • InHeadFC.parameters
        • InHeadFC.register_backward_hook
        • InHeadFC.register_buffer
        • InHeadFC.register_forward_hook
        • InHeadFC.register_forward_pre_hook
        • InHeadFC.register_full_backward_hook
        • InHeadFC.register_full_backward_pre_hook
        • InHeadFC.register_load_state_dict_post_hook
        • InHeadFC.register_load_state_dict_pre_hook
        • InHeadFC.register_module
        • InHeadFC.register_parameter
        • InHeadFC.register_state_dict_post_hook
        • InHeadFC.register_state_dict_pre_hook
        • InHeadFC.requires_grad_
        • InHeadFC.set_extra_state
        • InHeadFC.set_submodule
        • InHeadFC.share_memory
        • InHeadFC.state_dict
        • InHeadFC.to
        • InHeadFC.to_empty
        • InHeadFC.train
        • InHeadFC.type
        • InHeadFC.xpu
        • InHeadFC.zero_grad
      • OutHeadFC
        • OutHeadFC.forward
        • OutHeadFC.add_module
        • OutHeadFC.apply
        • OutHeadFC.bfloat16
        • OutHeadFC.buffers
        • OutHeadFC.children
        • OutHeadFC.compile
        • OutHeadFC.cpu
        • OutHeadFC.cuda
        • OutHeadFC.double
        • OutHeadFC.eval
        • OutHeadFC.extra_repr
        • OutHeadFC.float
        • OutHeadFC.get_buffer
        • OutHeadFC.get_extra_state
        • OutHeadFC.get_parameter
        • OutHeadFC.get_postprocessors
        • OutHeadFC.get_submodule
        • OutHeadFC.half
        • OutHeadFC.ipu
        • OutHeadFC.load_state_dict
        • OutHeadFC.modules
        • OutHeadFC.mtia
        • OutHeadFC.named_buffers
        • OutHeadFC.named_children
        • OutHeadFC.named_modules
        • OutHeadFC.named_parameters
        • OutHeadFC.parameters
        • OutHeadFC.postprocess
        • OutHeadFC.register_backward_hook
        • OutHeadFC.register_buffer
        • OutHeadFC.register_forward_hook
        • OutHeadFC.register_forward_pre_hook
        • OutHeadFC.register_full_backward_hook
        • OutHeadFC.register_full_backward_pre_hook
        • OutHeadFC.register_load_state_dict_post_hook
        • OutHeadFC.register_load_state_dict_pre_hook
        • OutHeadFC.register_module
        • OutHeadFC.register_parameter
        • OutHeadFC.register_state_dict_post_hook
        • OutHeadFC.register_state_dict_pre_hook
        • OutHeadFC.requires_grad_
        • OutHeadFC.set_extra_state
        • OutHeadFC.set_submodule
        • OutHeadFC.share_memory
        • OutHeadFC.state_dict
        • OutHeadFC.to
        • OutHeadFC.to_empty
        • OutHeadFC.train
        • OutHeadFC.type
        • OutHeadFC.xpu
        • OutHeadFC.zero_grad
      • InHeadConv
        • InHeadConv.forward
        • InHeadConv.add_module
        • InHeadConv.apply
        • InHeadConv.bfloat16
        • InHeadConv.buffers
        • InHeadConv.children
        • InHeadConv.compile
        • InHeadConv.cpu
        • InHeadConv.cuda
        • InHeadConv.double
        • InHeadConv.eval
        • InHeadConv.extra_repr
        • InHeadConv.float
        • InHeadConv.get_buffer
        • InHeadConv.get_extra_state
        • InHeadConv.get_parameter
        • InHeadConv.get_submodule
        • InHeadConv.half
        • InHeadConv.ipu
        • InHeadConv.load_state_dict
        • InHeadConv.modules
        • InHeadConv.mtia
        • InHeadConv.named_buffers
        • InHeadConv.named_children
        • InHeadConv.named_modules
        • InHeadConv.named_parameters
        • InHeadConv.parameters
        • InHeadConv.register_backward_hook
        • InHeadConv.register_buffer
        • InHeadConv.register_forward_hook
        • InHeadConv.register_forward_pre_hook
        • InHeadConv.register_full_backward_hook
        • InHeadConv.register_full_backward_pre_hook
        • InHeadConv.register_load_state_dict_post_hook
        • InHeadConv.register_load_state_dict_pre_hook
        • InHeadConv.register_module
        • InHeadConv.register_parameter
        • InHeadConv.register_state_dict_post_hook
        • InHeadConv.register_state_dict_pre_hook
        • InHeadConv.requires_grad_
        • InHeadConv.set_extra_state
        • InHeadConv.set_submodule
        • InHeadConv.share_memory
        • InHeadConv.state_dict
        • InHeadConv.to
        • InHeadConv.to_empty
        • InHeadConv.train
        • InHeadConv.type
        • InHeadConv.xpu
        • InHeadConv.zero_grad
      • OutHeadConv
        • OutHeadConv.forward
        • OutHeadConv.add_module
        • OutHeadConv.apply
        • OutHeadConv.bfloat16
        • OutHeadConv.buffers
        • OutHeadConv.children
        • OutHeadConv.compile
        • OutHeadConv.cpu
        • OutHeadConv.cuda
        • OutHeadConv.double
        • OutHeadConv.eval
        • OutHeadConv.extra_repr
        • OutHeadConv.float
        • OutHeadConv.get_buffer
        • OutHeadConv.get_extra_state
        • OutHeadConv.get_parameter
        • OutHeadConv.get_postprocessors
        • OutHeadConv.get_submodule
        • OutHeadConv.half
        • OutHeadConv.ipu
        • OutHeadConv.load_state_dict
        • OutHeadConv.modules
        • OutHeadConv.mtia
        • OutHeadConv.named_buffers
        • OutHeadConv.named_children
        • OutHeadConv.named_modules
        • OutHeadConv.named_parameters
        • OutHeadConv.parameters
        • OutHeadConv.postprocess
        • OutHeadConv.register_backward_hook
        • OutHeadConv.register_buffer
        • OutHeadConv.register_forward_hook
        • OutHeadConv.register_forward_pre_hook
        • OutHeadConv.register_full_backward_hook
        • OutHeadConv.register_full_backward_pre_hook
        • OutHeadConv.register_load_state_dict_post_hook
        • OutHeadConv.register_load_state_dict_pre_hook
        • OutHeadConv.register_module
        • OutHeadConv.register_parameter
        • OutHeadConv.register_state_dict_post_hook
        • OutHeadConv.register_state_dict_pre_hook
        • OutHeadConv.requires_grad_
        • OutHeadConv.set_extra_state
        • OutHeadConv.set_submodule
        • OutHeadConv.share_memory
        • OutHeadConv.state_dict
        • OutHeadConv.to
        • OutHeadConv.to_empty
        • OutHeadConv.train
        • OutHeadConv.type
        • OutHeadConv.xpu
        • OutHeadConv.zero_grad
      • InHeadConv1D
        • InHeadConv1D.add_module
        • InHeadConv1D.apply
        • InHeadConv1D.bfloat16
        • InHeadConv1D.buffers
        • InHeadConv1D.children
        • InHeadConv1D.compile
        • InHeadConv1D.cpu
        • InHeadConv1D.cuda
        • InHeadConv1D.double
        • InHeadConv1D.eval
        • InHeadConv1D.extra_repr
        • InHeadConv1D.float
        • InHeadConv1D.forward
        • InHeadConv1D.get_buffer
        • InHeadConv1D.get_extra_state
        • InHeadConv1D.get_parameter
        • InHeadConv1D.get_submodule
        • InHeadConv1D.half
        • InHeadConv1D.ipu
        • InHeadConv1D.load_state_dict
        • InHeadConv1D.modules
        • InHeadConv1D.mtia
        • InHeadConv1D.named_buffers
        • InHeadConv1D.named_children
        • InHeadConv1D.named_modules
        • InHeadConv1D.named_parameters
        • InHeadConv1D.parameters
        • InHeadConv1D.register_backward_hook
        • InHeadConv1D.register_buffer
        • InHeadConv1D.register_forward_hook
        • InHeadConv1D.register_forward_pre_hook
        • InHeadConv1D.register_full_backward_hook
        • InHeadConv1D.register_full_backward_pre_hook
        • InHeadConv1D.register_load_state_dict_post_hook
        • InHeadConv1D.register_load_state_dict_pre_hook
        • InHeadConv1D.register_module
        • InHeadConv1D.register_parameter
        • InHeadConv1D.register_state_dict_post_hook
        • InHeadConv1D.register_state_dict_pre_hook
        • InHeadConv1D.requires_grad_
        • InHeadConv1D.set_extra_state
        • InHeadConv1D.set_submodule
        • InHeadConv1D.share_memory
        • InHeadConv1D.state_dict
        • InHeadConv1D.to
        • InHeadConv1D.to_empty
        • InHeadConv1D.train
        • InHeadConv1D.type
        • InHeadConv1D.xpu
        • InHeadConv1D.zero_grad
      • OutHeadConv1D
        • OutHeadConv1D.add_module
        • OutHeadConv1D.apply
        • OutHeadConv1D.bfloat16
        • OutHeadConv1D.buffers
        • OutHeadConv1D.children
        • OutHeadConv1D.compile
        • OutHeadConv1D.cpu
        • OutHeadConv1D.cuda
        • OutHeadConv1D.double
        • OutHeadConv1D.eval
        • OutHeadConv1D.extra_repr
        • OutHeadConv1D.float
        • OutHeadConv1D.forward
        • OutHeadConv1D.get_buffer
        • OutHeadConv1D.get_extra_state
        • OutHeadConv1D.get_parameter
        • OutHeadConv1D.get_postprocessors
        • OutHeadConv1D.get_submodule
        • OutHeadConv1D.half
        • OutHeadConv1D.ipu
        • OutHeadConv1D.load_state_dict
        • OutHeadConv1D.modules
        • OutHeadConv1D.mtia
        • OutHeadConv1D.named_buffers
        • OutHeadConv1D.named_children
        • OutHeadConv1D.named_modules
        • OutHeadConv1D.named_parameters
        • OutHeadConv1D.parameters
        • OutHeadConv1D.postprocess
        • OutHeadConv1D.register_backward_hook
        • OutHeadConv1D.register_buffer
        • OutHeadConv1D.register_forward_hook
        • OutHeadConv1D.register_forward_pre_hook
        • OutHeadConv1D.register_full_backward_hook
        • OutHeadConv1D.register_full_backward_pre_hook
        • OutHeadConv1D.register_load_state_dict_post_hook
        • OutHeadConv1D.register_load_state_dict_pre_hook
        • OutHeadConv1D.register_module
        • OutHeadConv1D.register_parameter
        • OutHeadConv1D.register_state_dict_post_hook
        • OutHeadConv1D.register_state_dict_pre_hook
        • OutHeadConv1D.requires_grad_
        • OutHeadConv1D.set_extra_state
        • OutHeadConv1D.set_submodule
        • OutHeadConv1D.share_memory
        • OutHeadConv1D.state_dict
        • OutHeadConv1D.to
        • OutHeadConv1D.to_empty
        • OutHeadConv1D.train
        • OutHeadConv1D.type
        • OutHeadConv1D.xpu
        • OutHeadConv1D.zero_grad
      • InHeadConv2D
        • InHeadConv2D.add_module
        • InHeadConv2D.apply
        • InHeadConv2D.bfloat16
        • InHeadConv2D.buffers
        • InHeadConv2D.children
        • InHeadConv2D.compile
        • InHeadConv2D.cpu
        • InHeadConv2D.cuda
        • InHeadConv2D.double
        • InHeadConv2D.eval
        • InHeadConv2D.extra_repr
        • InHeadConv2D.float
        • InHeadConv2D.forward
        • InHeadConv2D.get_buffer
        • InHeadConv2D.get_extra_state
        • InHeadConv2D.get_parameter
        • InHeadConv2D.get_submodule
        • InHeadConv2D.half
        • InHeadConv2D.ipu
        • InHeadConv2D.load_state_dict
        • InHeadConv2D.modules
        • InHeadConv2D.mtia
        • InHeadConv2D.named_buffers
        • InHeadConv2D.named_children
        • InHeadConv2D.named_modules
        • InHeadConv2D.named_parameters
        • InHeadConv2D.parameters
        • InHeadConv2D.register_backward_hook
        • InHeadConv2D.register_buffer
        • InHeadConv2D.register_forward_hook
        • InHeadConv2D.register_forward_pre_hook
        • InHeadConv2D.register_full_backward_hook
        • InHeadConv2D.register_full_backward_pre_hook
        • InHeadConv2D.register_load_state_dict_post_hook
        • InHeadConv2D.register_load_state_dict_pre_hook
        • InHeadConv2D.register_module
        • InHeadConv2D.register_parameter
        • InHeadConv2D.register_state_dict_post_hook
        • InHeadConv2D.register_state_dict_pre_hook
        • InHeadConv2D.requires_grad_
        • InHeadConv2D.set_extra_state
        • InHeadConv2D.set_submodule
        • InHeadConv2D.share_memory
        • InHeadConv2D.state_dict
        • InHeadConv2D.to
        • InHeadConv2D.to_empty
        • InHeadConv2D.train
        • InHeadConv2D.type
        • InHeadConv2D.xpu
        • InHeadConv2D.zero_grad
      • OutHeadConv2D
        • OutHeadConv2D.add_module
        • OutHeadConv2D.apply
        • OutHeadConv2D.bfloat16
        • OutHeadConv2D.buffers
        • OutHeadConv2D.children
        • OutHeadConv2D.compile
        • OutHeadConv2D.cpu
        • OutHeadConv2D.cuda
        • OutHeadConv2D.double
        • OutHeadConv2D.eval
        • OutHeadConv2D.extra_repr
        • OutHeadConv2D.float
        • OutHeadConv2D.forward
        • OutHeadConv2D.get_buffer
        • OutHeadConv2D.get_extra_state
        • OutHeadConv2D.get_parameter
        • OutHeadConv2D.get_postprocessors
        • OutHeadConv2D.get_submodule
        • OutHeadConv2D.half
        • OutHeadConv2D.ipu
        • OutHeadConv2D.load_state_dict
        • OutHeadConv2D.modules
        • OutHeadConv2D.mtia
        • OutHeadConv2D.named_buffers
        • OutHeadConv2D.named_children
        • OutHeadConv2D.named_modules
        • OutHeadConv2D.named_parameters
        • OutHeadConv2D.parameters
        • OutHeadConv2D.postprocess
        • OutHeadConv2D.register_backward_hook
        • OutHeadConv2D.register_buffer
        • OutHeadConv2D.register_forward_hook
        • OutHeadConv2D.register_forward_pre_hook
        • OutHeadConv2D.register_full_backward_hook
        • OutHeadConv2D.register_full_backward_pre_hook
        • OutHeadConv2D.register_load_state_dict_post_hook
        • OutHeadConv2D.register_load_state_dict_pre_hook
        • OutHeadConv2D.register_module
        • OutHeadConv2D.register_parameter
        • OutHeadConv2D.register_state_dict_post_hook
        • OutHeadConv2D.register_state_dict_pre_hook
        • OutHeadConv2D.requires_grad_
        • OutHeadConv2D.set_extra_state
        • OutHeadConv2D.set_submodule
        • OutHeadConv2D.share_memory
        • OutHeadConv2D.state_dict
        • OutHeadConv2D.to
        • OutHeadConv2D.to_empty
        • OutHeadConv2D.train
        • OutHeadConv2D.type
        • OutHeadConv2D.xpu
        • OutHeadConv2D.zero_grad
      • InHeadConv3D
        • InHeadConv3D.add_module
        • InHeadConv3D.apply
        • InHeadConv3D.bfloat16
        • InHeadConv3D.buffers
        • InHeadConv3D.children
        • InHeadConv3D.compile
        • InHeadConv3D.cpu
        • InHeadConv3D.cuda
        • InHeadConv3D.double
        • InHeadConv3D.eval
        • InHeadConv3D.extra_repr
        • InHeadConv3D.float
        • InHeadConv3D.forward
        • InHeadConv3D.get_buffer
        • InHeadConv3D.get_extra_state
        • InHeadConv3D.get_parameter
        • InHeadConv3D.get_submodule
        • InHeadConv3D.half
        • InHeadConv3D.ipu
        • InHeadConv3D.load_state_dict
        • InHeadConv3D.modules
        • InHeadConv3D.mtia
        • InHeadConv3D.named_buffers
        • InHeadConv3D.named_children
        • InHeadConv3D.named_modules
        • InHeadConv3D.named_parameters
        • InHeadConv3D.parameters
        • InHeadConv3D.register_backward_hook
        • InHeadConv3D.register_buffer
        • InHeadConv3D.register_forward_hook
        • InHeadConv3D.register_forward_pre_hook
        • InHeadConv3D.register_full_backward_hook
        • InHeadConv3D.register_full_backward_pre_hook
        • InHeadConv3D.register_load_state_dict_post_hook
        • InHeadConv3D.register_load_state_dict_pre_hook
        • InHeadConv3D.register_module
        • InHeadConv3D.register_parameter
        • InHeadConv3D.register_state_dict_post_hook
        • InHeadConv3D.register_state_dict_pre_hook
        • InHeadConv3D.requires_grad_
        • InHeadConv3D.set_extra_state
        • InHeadConv3D.set_submodule
        • InHeadConv3D.share_memory
        • InHeadConv3D.state_dict
        • InHeadConv3D.to
        • InHeadConv3D.to_empty
        • InHeadConv3D.train
        • InHeadConv3D.type
        • InHeadConv3D.xpu
        • InHeadConv3D.zero_grad
      • OutHeadConv3D
        • OutHeadConv3D.add_module
        • OutHeadConv3D.apply
        • OutHeadConv3D.bfloat16
        • OutHeadConv3D.buffers
        • OutHeadConv3D.children
        • OutHeadConv3D.compile
        • OutHeadConv3D.cpu
        • OutHeadConv3D.cuda
        • OutHeadConv3D.double
        • OutHeadConv3D.eval
        • OutHeadConv3D.extra_repr
        • OutHeadConv3D.float
        • OutHeadConv3D.forward
        • OutHeadConv3D.get_buffer
        • OutHeadConv3D.get_extra_state
        • OutHeadConv3D.get_parameter
        • OutHeadConv3D.get_postprocessors
        • OutHeadConv3D.get_submodule
        • OutHeadConv3D.half
        • OutHeadConv3D.ipu
        • OutHeadConv3D.load_state_dict
        • OutHeadConv3D.modules
        • OutHeadConv3D.mtia
        • OutHeadConv3D.named_buffers
        • OutHeadConv3D.named_children
        • OutHeadConv3D.named_modules
        • OutHeadConv3D.named_parameters
        • OutHeadConv3D.parameters
        • OutHeadConv3D.postprocess
        • OutHeadConv3D.register_backward_hook
        • OutHeadConv3D.register_buffer
        • OutHeadConv3D.register_forward_hook
        • OutHeadConv3D.register_forward_pre_hook
        • OutHeadConv3D.register_full_backward_hook
        • OutHeadConv3D.register_full_backward_pre_hook
        • OutHeadConv3D.register_load_state_dict_post_hook
        • OutHeadConv3D.register_load_state_dict_pre_hook
        • OutHeadConv3D.register_module
        • OutHeadConv3D.register_parameter
        • OutHeadConv3D.register_state_dict_post_hook
        • OutHeadConv3D.register_state_dict_pre_hook
        • OutHeadConv3D.requires_grad_
        • OutHeadConv3D.set_extra_state
        • OutHeadConv3D.set_submodule
        • OutHeadConv3D.share_memory
        • OutHeadConv3D.state_dict
        • OutHeadConv3D.to
        • OutHeadConv3D.to_empty
        • OutHeadConv3D.train
        • OutHeadConv3D.type
        • OutHeadConv3D.xpu
        • OutHeadConv3D.zero_grad
      • LossStd
        • LossStd.forward
        • LossStd.add_module
        • LossStd.apply
        • LossStd.bfloat16
        • LossStd.buffers
        • LossStd.children
        • LossStd.compile
        • LossStd.cpu
        • LossStd.cuda
        • LossStd.double
        • LossStd.eval
        • LossStd.extra_repr
        • LossStd.float
        • LossStd.get_buffer
        • LossStd.get_extra_state
        • LossStd.get_parameter
        • LossStd.get_submodule
        • LossStd.half
        • LossStd.ipu
        • LossStd.load_state_dict
        • LossStd.modules
        • LossStd.mtia
        • LossStd.named_buffers
        • LossStd.named_children
        • LossStd.named_modules
        • LossStd.named_parameters
        • LossStd.parameters
        • LossStd.register_backward_hook
        • LossStd.register_buffer
        • LossStd.register_forward_hook
        • LossStd.register_forward_pre_hook
        • LossStd.register_full_backward_hook
        • LossStd.register_full_backward_pre_hook
        • LossStd.register_load_state_dict_post_hook
        • LossStd.register_load_state_dict_pre_hook
        • LossStd.register_module
        • LossStd.register_parameter
        • LossStd.register_state_dict_post_hook
        • LossStd.register_state_dict_pre_hook
        • LossStd.requires_grad_
        • LossStd.set_extra_state
        • LossStd.set_submodule
        • LossStd.share_memory
        • LossStd.state_dict
        • LossStd.to
        • LossStd.to_empty
        • LossStd.train
        • LossStd.type
        • LossStd.xpu
        • LossStd.zero_grad
      • MGEloss
        • MGEloss.forward
        • MGEloss.add_module
        • MGEloss.apply
        • MGEloss.bfloat16
        • MGEloss.buffers
        • MGEloss.children
        • MGEloss.compile
        • MGEloss.cpu
        • MGEloss.cuda
        • MGEloss.double
        • MGEloss.eval
        • MGEloss.extra_repr
        • MGEloss.float
        • MGEloss.get_buffer
        • MGEloss.get_extra_state
        • MGEloss.get_parameter
        • MGEloss.get_submodule
        • MGEloss.half
        • MGEloss.ipu
        • MGEloss.load_state_dict
        • MGEloss.modules
        • MGEloss.mtia
        • MGEloss.named_buffers
        • MGEloss.named_children
        • MGEloss.named_modules
        • MGEloss.named_parameters
        • MGEloss.parameters
        • MGEloss.register_backward_hook
        • MGEloss.register_buffer
        • MGEloss.register_forward_hook
        • MGEloss.register_forward_pre_hook
        • MGEloss.register_full_backward_hook
        • MGEloss.register_full_backward_pre_hook
        • MGEloss.register_load_state_dict_post_hook
        • MGEloss.register_load_state_dict_pre_hook
        • MGEloss.register_module
        • MGEloss.register_parameter
        • MGEloss.register_state_dict_post_hook
        • MGEloss.register_state_dict_pre_hook
        • MGEloss.requires_grad_
        • MGEloss.set_extra_state
        • MGEloss.set_submodule
        • MGEloss.share_memory
        • MGEloss.state_dict
        • MGEloss.to
        • MGEloss.to_empty
        • MGEloss.train
        • MGEloss.type
        • MGEloss.xpu
        • MGEloss.zero_grad
      • Generator
        • Generator.generate
        • Generator.info_attributes_req
        • Generator.print_results_gen
      • GeneratorSampler
        • GeneratorSampler.generate
        • GeneratorSampler.generate_z
        • GeneratorSampler.summary_sampling_types
      • LocalSensitivity
        • LocalSensitivity.calculate
        • LocalSensitivity.plot
      • GlobalSensitivity
        • GlobalSensitivity.calculate
        • GlobalSensitivity.plot
    • aixd.sampler
      • GeneratorCallback
      • SamplingEngine
        • SamplingEngine.deserialise
        • SamplingEngine.reset_states
        • SamplingEngine.sample
        • SamplingEngine.update
      • AgnosticSamplingEngine
        • AgnosticSamplingEngine.update
        • AgnosticSamplingEngine.deserialise
        • AgnosticSamplingEngine.reset_states
        • AgnosticSamplingEngine.sample
      • AdaptiveSamplingEngine
        • AdaptiveSamplingEngine.update
        • AdaptiveSamplingEngine.deserialise
        • AdaptiveSamplingEngine.reset_states
        • AdaptiveSamplingEngine.sample
      • RandomSamplingEngine
        • RandomSamplingEngine.sample
        • RandomSamplingEngine.deserialise
        • RandomSamplingEngine.reset_states
        • RandomSamplingEngine.update
      • GridSamplingEngine
        • GridSamplingEngine.sample
        • GridSamplingEngine.deserialise
        • GridSamplingEngine.reset_states
        • GridSamplingEngine.update
      • SobolSamplingEngine
        • SobolSamplingEngine.reset_states
        • SobolSamplingEngine.sample
        • SobolSamplingEngine.deserialise
        • SobolSamplingEngine.update
      • LHCSamplingEngine
        • LHCSamplingEngine.reset_states
        • LHCSamplingEngine.sample
        • LHCSamplingEngine.deserialise
        • LHCSamplingEngine.update
      • BayesOptSamplingEngine
        • BayesOptSamplingEngine.reset_states
        • BayesOptSamplingEngine.sample
        • BayesOptSamplingEngine.update
        • BayesOptSamplingEngine.deserialise
      • Arithmetic
        • Arithmetic.is_differentiable
        • Arithmetic.evaluate
        • Arithmetic.print_all
      • Boolean
        • Boolean.is_differentiable
        • Boolean.evaluate
        • Boolean.print_all
      • Constant
        • Constant.evaluate
        • Constant.is_differentiable
        • Constant.print_all
      • Add
        • Add.evaluate
        • Add.is_differentiable
        • Add.print_all
      • Multiply
        • Multiply.evaluate
        • Multiply.is_differentiable
        • Multiply.print_all
      • Subtract
        • Subtract.evaluate
        • Subtract.is_differentiable
        • Subtract.print_all
      • Divide
        • Divide.evaluate
        • Divide.is_differentiable
        • Divide.print_all
      • LessThan
        • LessThan.evaluate
        • LessThan.is_differentiable
        • LessThan.print_all
      • LessOrEqual
        • LessOrEqual.evaluate
        • LessOrEqual.is_differentiable
        • LessOrEqual.print_all
      • GreaterThan
        • GreaterThan.evaluate
        • GreaterThan.is_differentiable
        • GreaterThan.print_all
      • GreaterOrEqual
        • GreaterOrEqual.evaluate
        • GreaterOrEqual.is_differentiable
        • GreaterOrEqual.print_all
      • Log
        • Log.evaluate
        • Log.is_differentiable
        • Log.print_all
      • Exp
        • Exp.evaluate
        • Exp.is_differentiable
        • Exp.print_all
      • Pow
        • Pow.evaluate
        • Pow.is_differentiable
        • Pow.print_all
      • Not
        • Not.evaluate
        • Not.is_differentiable
        • Not.print_all
      • And
        • And.evaluate
        • And.is_differentiable
        • And.print_all
      • Or
        • Or.evaluate
        • Or.is_differentiable
        • Or.print_all
      • XOr
        • XOr.evaluate
        • XOr.is_differentiable
        • XOr.print_all
      • Equal
        • Equal.evaluate
        • Equal.is_differentiable
        • Equal.print_all
      • Negative
        • Negative.evaluate
        • Negative.is_differentiable
        • Negative.print_all
      • CastBooleanToConstant
        • CastBooleanToConstant.evaluate
        • CastBooleanToConstant.is_differentiable
        • CastBooleanToConstant.print_all
      • Reducer
        • Reducer.evaluate
        • Reducer.is_differentiable
      • Sum
        • Sum.evaluate
        • Sum.is_differentiable
      • Mean
        • Mean.evaluate
        • Mean.is_differentiable
      • Std
        • Std.evaluate
        • Std.is_differentiable
      • Var
        • Var.evaluate
        • Var.is_differentiable
      • All
        • All.evaluate
        • All.is_differentiable
      • Any
        • Any.evaluate
        • Any.is_differentiable
      • SamplesGenerator
        • SamplesGenerator.fit
        • SamplesGenerator.generate
        • SamplesGenerator.reset_states
      • Strategy
        • Strategy.fit
        • Strategy.reset_states
        • Strategy.sample
        • Strategy.update
      • UniformStrategy
        • UniformStrategy.sample
        • UniformStrategy.fit
        • UniformStrategy.reset_states
        • UniformStrategy.update
      • QuantileStrategy
        • QuantileStrategy.fit
        • QuantileStrategy.reset_states
        • QuantileStrategy.sample
        • QuantileStrategy.update
      • KernelDensityStrategy
        • KernelDensityStrategy.fit
        • KernelDensityStrategy.reset_states
        • KernelDensityStrategy.sample
        • KernelDensityStrategy.update
    • aixd.visualisation
      • Plotter
        • Plotter.attributes_obs_vs_pred
        • Plotter.available_block_names
        • Plotter.contours2d
        • Plotter.correlation
        • Plotter.distrib_attributes
        • Plotter.distrib_attributes2d
        • Plotter.evaluate_generation
        • Plotter.evaluate_training
        • Plotter.generation_scatter
        • Plotter.kde_distribution_attributes
        • Plotter.parallel_coordinates
        • Plotter.performance_summary
    • aixd.utils
      • Logger
        • Logger.from_config
        • Logger.get_child
        • Logger.get_name
        • Logger.set_verbosity
        • Logger.addFilter
        • Logger.addHandler
        • Logger.callHandlers
        • Logger.critical
        • Logger.debug
        • Logger.error
        • Logger.exception
        • Logger.fatal
        • Logger.filter
        • Logger.findCaller
        • Logger.getChild
        • Logger.getChildren
        • Logger.getEffectiveLevel
        • Logger.handle
        • Logger.hasHandlers
        • Logger.info
        • Logger.isEnabledFor
        • Logger.log
        • Logger.makeRecord
        • Logger.removeFilter
        • Logger.removeHandler
        • Logger.setLevel
        • Logger.warn
        • Logger.warning
  • API Reference
  • AI-eXtended Design
  • aixd.mlmodel
  • InHeadFC
  • InHeadFC.register_module

InHeadFC.register_module

InHeadFC.register_module(name: str, module: Module | None) → None[source]

Alias for add_module().

previous

InHeadFC.register_load_state_dict_pre_hook

next

InHeadFC.register_parameter
On this page
  • InHeadFC.register_module()

© Copyright 2020-2026 ETH Zurich & Swiss Data Science Center.

Created using Sphinx 7.4.7.

Built with the PyData Sphinx Theme 0.16.1.