/usr/local/lib64/python3.6/site-packages/caffe2/python/layers/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/layers/__pycache__/dropout.cpython-36.pyc (1727B)
3
Eg @ s, d dl mZ d dlmZ G dd deZdS ) )schema)
ModelLayerc s> e Zd Zd fdd Zdd Zdd Zd
d Zdd
Z ZS )Dropoutdropout ?Fc st t t| j|||f| t|tjs,td|dkr<|dk sHtd| |j | _| jj | j
d || _|| _d S )NzIncorrect input typer g ?z,Expected 0 <= ratio < 1, but got ratio of %soutput)
superr __init__
isinstancer ZScalarAssertionErrorZclone_schema
output_schema set_valueZget_next_blob_referencedropout_for_evalratio)selfmodelinput_recordnamer r kwargs) __class__ H/usr/local/lib64/python3.6/site-packages/caffe2/python/layers/dropout.pyr
s
zDropout.__init__c C s8 | j j }| jj |jdg }|j||| j|d d S )NZd_mask)r is_test)r Zfield_blobsr ZNextScopedBlobr r )r netr Z
input_blobZoutput_blobsr r r _add_ops! s
zDropout._add_opsc C s | j |dd d S )NF)r )r )r r r r r
add_train_ops+ s zDropout.add_train_opsc C s | j || j d d S )N)r )r r )r r r r r add_eval_ops. s zDropout.add_eval_opsc C s | j | d S )N)r )r r r r r add_ops1 s zDropout.add_ops)r r F) __name__
__module____qualname__r r r r r
__classcell__r r )r r r s
r N)Z
caffe2.pythonr Zcaffe2.python.layers.layersr r r r r r
s