Name	Size	Mode	Actions
adaptive_weight.cpython-36.pyc	5287	0644	editdlrm
add_bias.cpython-36.pyc	1479	0644	editdlrm
arc_cosine_feature_map.cpython-36.pyc	5427	0644	editdlrm
batch_huber_loss.cpython-36.pyc	2328	0644	editdlrm
batch_lr_loss.cpython-36.pyc	5565	0644	editdlrm
batch_mse_loss.cpython-36.pyc	1839	0644	editdlrm
batch_normalization.cpython-36.pyc	2779	0644	editdlrm
batch_sigmoid_cross_entropy_loss.cpython-36.pyc	1542	0644	editdlrm
batch_softmax_loss.cpython-36.pyc	2877	0644	editdlrm
blob_weighted_sum.cpython-36.pyc	2420	0644	editdlrm
bpr_loss.cpython-36.pyc	1405	0644	editdlrm
bucket_weighted.cpython-36.pyc	2079	0644	editdlrm
build_index.cpython-36.pyc	2068	0644	editdlrm
concat.cpython-36.pyc	4063	0644	editdlrm
constant_weight.cpython-36.pyc	1687	0644	editdlrm
conv.cpython-36.pyc	3244	0644	editdlrm
dropout.cpython-36.pyc	1727	0644	editdlrm
fc.cpython-36.pyc	6159	0644	editdlrm
fc_without_bias.cpython-36.pyc	1891	0644	editdlrm
fc_with_bootstrap.cpython-36.pyc	8653	0644	editdlrm
feature_sparse_to_dense.cpython-36.pyc	4571	0644	editdlrm
functional.cpython-36.pyc	2935	0644	editdlrm
gather_record.cpython-36.pyc	2886	0644	editdlrm
homotopy_weight.cpython-36.pyc	3137	0644	editdlrm
label_smooth.cpython-36.pyc	2646	0644	editdlrm
last_n_window_collector.cpython-36.pyc	1779	0644	editdlrm
layers.cpython-36.pyc	13722	0644	editdlrm
layer_normalization.cpython-36.pyc	3128	0644	editdlrm
margin_rank_loss.cpython-36.pyc	1808	0644	editdlrm
merge_id_lists.cpython-36.pyc	1984	0644	editdlrm
pairwise_similarity.cpython-36.pyc	2365	0644	editdlrm
position_weighted.cpython-36.pyc	1985	0644	editdlrm
random_fourier_features.cpython-36.pyc	2550	0644	editdlrm
reservoir_sampling.cpython-36.pyc	2015	0644	editdlrm
sampling_train.cpython-36.pyc	2073	0644	editdlrm
sampling_trainable_mixin.cpython-36.pyc	1873	0644	editdlrm
select_record_by_context.cpython-36.pyc	2568	0644	editdlrm
semi_random_features.cpython-36.pyc	4557	0644	editdlrm
sparse_dropout_with_replacement.cpython-36.pyc	2379	0644	editdlrm
sparse_feature_hash.cpython-36.pyc	2890	0644	editdlrm
sparse_itemwise_dropout_with_replacement.cpython-36.pyc	2447	0644	editdlrm
sparse_lookup.cpython-36.pyc	12514	0644	editdlrm
split.cpython-36.pyc	1924	0644	editdlrm
tags.cpython-36.pyc	2784	0644	editdlrm
uniform_sampling.cpython-36.pyc	2102	0644	editdlrm
__init__.cpython-36.pyc	1048	0644	editdlrm

3 Eg��@sfdd
lm
Z

ddlmZ
ddlmZ
ddlZddlm Z
ddl Z e je�
Z dd�ZGdd �d e�ZdS) �)
�schema)
� ModelLayer)
� viewitemsN)
�defaultdictc
s�tt
�
}
d
�x�|D]�}|jjd
}t|d�r�t|jd�r�t|jjd�r�t|jjjt �r�x6|jjjj �D]$\}}|
|j�f
dd�|D�
�

qfW�|7�qW|
j�r�t |
�
SdS)Nr
�metadata� feature_specs�feature_to_indexc
sg|]}
�|
�qS�r )�.0�vi)
� start_posr �G/usr/local/lib64/python3.6/site-packages/caffe2/python/layers/concat.py� sz5get_concatenated_feature_to_index..) r�listZdtype�shape�hasattrrr� isinstancer�dict�items�extend�keys)Zblobs_to_concatZconcat_feature_to_indexZscalarZnum_dims�
k�
vr )
rr �!get_concatenated_feature_to_indexs











rcs*eZ
dZd
Zd �f
dd� Zdd�Z�ZS) �Concata� Construct Concat layer Assume that first dimension is batch, Example: embedding_dim = 64 input_record = self.new_record(schema.Struct( ('input1', schema.Scalar((np.float32, (embedding_dim, )))), ('input2', schema.Scalar((np.float32, (embedding_dim, )))), ('input3', schema.Scalar((np.float32, (embedding_dim, )))), )) output = self.model.Concat(input_record) self.assertEqual( schema.Scalar((np.float32, ((len(input_record.fields) * embedding_dim, )))), output ) # Note that in Concat layer we assume first dimension is batch. # so input is B * embedding_dim # add_axis=1 make it B * 1 * embedding_dim # Concat on axis=1 make it B * N * embedding_dim output = self.model.Concat(input_record, axis=1, add_axis=1) self.assertEqual( schema.Scalar((np.float32, ((len(input_record.fields), embedding_dim)))), output ) �
r
�concatcs�
tt
|�j|
||f|�

||_||_|d
ko2|dks>td�
�
t|tj�sXtdj |�
�
�
g}xrt |j�
D]d\}} t| tj�s�tdj || ��
�
t | j�j�
} |r�| j|dd�
t| �
|ks�td�
�
|j| �

qhWtjdt|�
�

|d
k�
r tj|d
|jd�
g
�|_dSd
}xL|D]D} || |d7}d
| |d<| |d
k�
std j | |d
��
�
�
qW|d
}|||d<tjd t|�
�

tjtj|f|jd�
�|_|jj�} t| �
}|�
r�tjtj|d�
d�
}|jj|�

dS) Nr
rzIt's not allowed to add axis=0z8Incorrect input type. Expected Struct, but received: {0}z9Incorrect input type for {}. Expected Scalar, but got: {}z:Concat expects that limited dimensions of the input tensorzConcat Layer input shapes: �outputz0Shapes {0} and {1} are not compatible for ConcatzConcat Layer output_dims: )
r)
r) �superr�__init__�axis�add_axis�AssertionErrorrr�Struct�formatr�fieldsZScalarr� field_typer�insert�len�append�logger�info�strZfrom_blob_listZget_next_blob_reference� output_schema�npZfloat32�valuesrZMetadataZFeatureSpecZset_metadata)�self�model�input_recordr r!�name�kwargsZshapes� field_namer&rZ concat_dimZoutput_dimsZrecord_to_concatZconcated_feature_to_indexr)
� __class__r r rCs\



































zConcat.__init__cCs<|
j|j
j�|jj�d
|jj�d
dg|j|jd�
dS)Nr
Z_concat_dims)r r!)rr2Zfield_blobsr-r r!)r0�netr r r �add_ops�s



zConcat.add_ops)rr
r)�__name__� __module__�__qualname__�__doc__rr8� __classcell__r r )
r6r r#s
s