o
    z“©iY  ã                   @   s,   d dl Z ddlmZ G dd„ de jjƒZdS )é    Né   )Úconv1x1c                       s0   e Zd ZdZd	def‡ fdd„Zdd„ Z‡  ZS )
ÚCombinez*Combine information from skip connections.Úcaté   Ú
dimensionsc                    s<   t ƒ  ¡  t|||d| _|dv sJ d|› dƒ‚|| _d S )N)r   )r   ÚsumzMethod z not recognized.)ÚsuperÚ__init__r   ÚConv_0Úmethod)ÚselfÚin_chÚout_chr   r   ©Ú	__class__© úO/home/ubuntu/.local/lib/python3.10/site-packages/score_models/layers/combine.pyr
      s   

zCombine.__init__c                 C   s<   |   |¡}| jdkrtj||gddS | jdkr|| S d S )Nr   r   )Údimr   )r   r   Útorchr   )r   ÚxÚyÚhr   r   r   Úforward   s   


ÿzCombine.forward)r   r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úintr
   r   Ú__classcell__r   r   r   r   r      s    r   )r   Úconv_layersr   ÚnnÚModuler   r   r   r   r   Ú<module>   s    