Initial commit
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
//! # Multimodal Fusion Implementation
|
||||
//!
|
||||
//! Unified multimodal representation learning and fusion layer.
|
||||
|
||||
mod config;
|
||||
mod encoder;
|
||||
mod modality;
|
||||
mod strategies;
|
||||
mod transformer;
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests;
|
||||
|
||||
// Re-export configuration types
|
||||
pub use config::{
|
||||
AttentionFusionConfig, BottleneckFusionConfig, ContrastiveLearningConfig,
|
||||
CrossModalAttentionConfig, FusionConfig, GatedFusionConfig, HierarchicalFusionConfig,
|
||||
MultimodalFusionConfig, SequentialFusionConfig, TensorFusionConfig,
|
||||
};
|
||||
|
||||
// Re-export encoder
|
||||
pub use encoder::ModalityEncoder;
|
||||
|
||||
// Re-export modality fusion
|
||||
pub use modality::{CrossModalFusionLayer, FusionMLP, ModalityFusion};
|
||||
|
||||
// Re-export strategies
|
||||
pub use strategies::{
|
||||
AttentionBasedFusion, BilinearFusion, FusionOutput, FusionStats, GatedFusion,
|
||||
HierarchicalFusion, ModalityFusionStrategy, MultimodalBottleneckFusion,
|
||||
MultimodalContrastiveLearning, SequentialFusion, TensorFusionNetwork,
|
||||
};
|
||||
|
||||
// Re-export transformer
|
||||
pub use transformer::MultimodalFusionTransformer;
|
||||
Reference in New Issue
Block a user