Name	Size	Mode	Actions
activation.cpython-36.pyc	47635	0644	editdlrm
adaptive.cpython-36.pyc	9725	0644	editdlrm
batchnorm.cpython-36.pyc	31039	0644	editdlrm
channelshuffle.cpython-36.pyc	1901	0644	editdlrm
container.cpython-36.pyc	27970	0644	editdlrm
conv.cpython-36.pyc	57789	0644	editdlrm
distance.cpython-36.pyc	3684	0644	editdlrm
dropout.cpython-36.pyc	10310	0644	editdlrm
flatten.cpython-36.pyc	5813	0644	editdlrm
fold.cpython-36.pyc	12807	0644	editdlrm
instancenorm.cpython-36.pyc	18808	0644	editdlrm
lazy.cpython-36.pyc	11690	0644	editdlrm
linear.cpython-36.pyc	10188	0644	editdlrm
loss.cpython-36.pyc	91465	0644	editdlrm
module.cpython-36.pyc	66530	0644	editdlrm
normalization.cpython-36.pyc	11360	0644	editdlrm
padding.cpython-36.pyc	22217	0644	editdlrm
pixelshuffle.cpython-36.pyc	4398	0644	editdlrm
pooling.cpython-36.pyc	53203	0644	editdlrm
rnn.cpython-36.pyc	44653	0644	editdlrm
sparse.cpython-36.pyc	20956	0644	editdlrm
transformer.cpython-36.pyc	20737	0644	editdlrm
upsampling.cpython-36.pyc	10780	0644	editdlrm
utils.cpython-36.pyc	2528	0644	editdlrm
_functions.cpython-36.pyc	5441	0644	editdlrm
__init__.cpython-36.pyc	5236	0644	editdlrm

3 �Eg]�@sHdd
lm
Z

ddlmZ
ddlmZ
Gdd�de
�ZGdd �d e
�Zd S)�
)
�Module�)
� functional�)
�TensorcsXeZ
dZUd
ZdddgZeee deeedd ��f
d d� Z eeed�d d�Z�Z S)�PairwiseDistancea Computes the pairwise distance between vectors :math:`v_1`, :math:`v_2` using the p-norm: .. math :: \Vert x \Vert _p = \left( \sum_{i=1}^n \vert x_i \vert ^ p \right) ^ {1/p}. Args: p (real): the norm degree. Default: 2 eps (float, optional): Small value to avoid division by zero. Default: 1e-6 keepdim (bool, optional): Determines whether or not to keep the vector dimension. Default: False Shape: - Input1: :math:`(N, D)` or :math:`(D)` where `N = batch dimension` and `D = vector dimension` - Input2: :math:`(N, D)` or :math:`(D)`, same shape as the Input1 - Output: :math:`(N)` or :math:`()` based on input dimension. If :attr:`keepdim` is ``True``, then :math:`(N, 1)` or :math:`(1)` based on input dimension. Examples:: >>> pdist = nn.PairwiseDistance(p=2) >>> input1 = torch.randn(100, 128) >>> input2 = torch.randn(100, 128) >>> output = pdist(input1, input2) �norm�eps�keepdim�@�����ư>FN)�
pr r �returncs$tt
|�j�
|
|_||_||_dS)
N)�superr�__init__rr r )�selfr r r )
� __class__��E/usr/local/lib64/python3.6/site-packages/torch/nn/modules/distance.pyr$s



zPairwiseDistance.__init__)�x1�x2rcCstj
|
||j|j|j�S)
N)�
FZpairwise_distancerr r )rrrrrr�forward*s
zPairwiseDistance.forward)rrF)�__name__� __module__�__qualname__�__doc__� __constants__�floatrr �boolr rrr� __classcell__rr)
rrrs 



rcsPeZ
dZUd
ZddgZeed eedd��f
dd � Z e e e d �dd�Z�ZS)�CosineSimilaritya�Returns cosine similarity between :math:`x_1` and :math:`x_2`, computed along `dim`. .. math :: \text{similarity} = \dfrac{x_1 \cdot x_2}{\max(\Vert x_1 \Vert _2 \cdot \Vert x_2 \Vert _2, \epsilon)}. Args: dim (int, optional): Dimension where cosine similarity is computed. Default: 1 eps (float, optional): Small value to avoid division by zero. Default: 1e-8 Shape: - Input1: :math:`(\ast_1, D, \ast_2)` where D is at position `dim` - Input2: :math:`(\ast_1, D, \ast_2)`, same number of dimensions as x1, matching x1 size at dimension `dim`, and broadcastable with x1 at other dimensions. - Output: :math:`(\ast_1, \ast_2)` Examples:: >>> input1 = torch.randn(100, 128) >>> input2 = torch.randn(100, 128) >>> cos = nn.CosineSimilarity(dim=1, eps=1e-6) >>> output = cos(input1, input2) �dimr r
�:�0�yE>N)r"r rcstt
|�j�
|
|_||_dS)
N)rr!rr"r )rr"r )
rrrrGs


zCosineSimilarity.__init__)rrrcCstj
|
||j|j�S)
N)rZcosine_similarityr"r )rrrrrrrLs
zCosineSimilarity.forward)r
r#) rrrrr�intr"rr rrrr rr)
rrr!.s 


r!N) �moduler�rrZtorchrrr!rrrr�
s
'