
    h	                     V    S r SSKrSSKJr  SSKJrJr   " S S\R                  5      rg)zC
Unified Consciousness Layer - Integrates Eden's 6 existing layers
    N)	FIBONACCIINV_PHIc                   6   ^  \ rS rSrSrSU 4S jjrS rSrU =r$ )UnifiedConsciousnessLayer   zU
Integrates all 6 of Eden's existing consciousness layers
into one unified awareness
c                   > [         TU ]  5         Xl        X l        [        R
                  " USSS9U l        X-  n[        R                  " [        R                  " XC5      [        R                  " U5      [        R                  " 5       [        R                  " X35      [        R                  " U5      [        R                  " 5       [        R                  " X25      5      U l        [        R                  " [        R                  " US5      [        R                  " 5       [        R                  " SS5      [        R                  " 5       5      U l        g )Nr   T)	embed_dim	num_headsbatch_first       )super__init__
num_layerslayer_output_sizennMultiheadAttentioncross_attention
SequentialLinear	LayerNormGELUintegrationTanhSigmoidresonance_head)selfr   r   unified_sizetotal_input	__class__s        ;/home/james-whalen/eden_hybrid_phi/unified_consciousness.pyr   "UnifiedConsciousnessLayer.__init__   s    $!2  "44' 
 !4==IIk0LL&GGIIIl1LL&GGIIIl6
 !mmIIlB'GGIIIb!JJL	
    c           	      ^   [         R                  " USS9nU R                  X"U5      u  p4UR                  SS9nU R	                  U5      nU R                  S   " U R                  S   " U R                  S   " U R                  S   " U5      5      5      5      nU R                  U5      nUUUS.$ )zi
Args:
    layer_outputs: list of 6 tensors [batch, output_size]
Returns:
    unified output + resonance
r   )dim)	start_dim      r   )unified_output	resonanceattention_weights)torchstackr   flattenr   r   )	r   layer_outputsstackedattendedr+   	flattenedunifiedresonance_inputr*   s	            r!   forward!UnifiedConsciousnessLayer.forward/   s     ++m3 '+&:&:g'
#
 $$q$1	 ""9- **1-d.>.>q.AQ 0 0 3I >?/
  ''8	 &"!2
 	
r#   )r   r   r   r   r   )   @      )	__name__
__module____qualname____firstlineno____doc__r   r5   __static_attributes____classcell__)r    s   @r!   r   r      s    

B
 
r#   r   )	r>   r,   torch.nnr   phi_constantsr   r   Moduler    r#   r!   <module>rE      s'      ,F
		 F
r#   