InPeerReview/RemoteSensingChangeDetection-RSCD.HA2F
2
1a
2 8��g� � @ s4 d dl mZ d3eed�dd�ZG dd� dej�Zd S ) � )�nn� F)� drop_prob�trainingc C s` |dks |s| S d| }| j d fd| jd }| �|��|�}|dkrT|�|� | | }|S )Nr � r )r )�shape�ndim� new_empty�4bernoulli_�div_)�xr r Z keep_probr Z
random_tensor�output� r �FC:\Users\82607\Desktop\code\ChangeViT-master\model\layers\drop_path.py� drop_path s 5r c s* e Zd ZdZd� fdd� Zdd� Z� ZS )�DropPathzXDrop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).Nc s t t| ��� || _d S �N)�superr �__init__r )�selfr �� __class__r r r s zDropPath.__init__c C s t || j| j�S r )r r r )r r r r r �forward! s zDropPath.forward)N)�__name__�6__module__� __qualname__�__doc__r r �
__classcell__r r r r r s r N)r F)�torchr �float�boolr �Moduler r r r r �<module> s