o Qe@@sVddlZddlmZmZddlmZddlmZmZddlm Z m Z gZ d d d Z dS) N)check_variable_and_dtype check_type) LayerHelper)_C_ops _legacy_C_ops)in_dygraph_mode_in_legacy_dygraph@ư>Fc Cst|dttfdt|dtdt|dtdtr@t||}|dkr6tjj j j |g|j d}t ||}t||dd|dStrlt||}|dkr_tjj j j |g|j d}t||}t|d dd|d|dd St|d d d gdt|d d d gdt||}|dkr|jj|j d}tjdg||j d}t ||}td|d}d||dd} |j|j d} |jdd|id| i| d| S)a It computes the pairwise distance between two vectors. The distance is calculated by p-oreder norm: .. math:: \Vert x \Vert _p = \left( \sum_{i=1}^n \vert x_i \vert ^ p \right) ^ {1/p}. Parameters: x (Tensor): Tensor, shape is :math:`[N, D]` or :math:`[D]`, where :math:`N` is batch size, :math:`D` is the dimension of vector. Available dtype is float32, float64. y (Tensor): Tensor, shape is :math:`[N, D]` or :math:`[D]`, where :math:`N` is batch size, :math:`D` is the dimension of vector. Available dtype is float32, float64. p (float, optional): The order of norm. Default: :math:`2.0`. epsilon (float, optional): Add small value to avoid division by zero. Default: :math:`1e-6`. keepdim (bool, optional): Whether to reserve the reduced dimension in the output Tensor. The result tensor is one dimension less than the result of ``|x-y|`` unless :attr:`keepdim` is True. Default: False. name (str, optional): For details, please refer to :ref:`api_guide_Name`. Generally, no setting is required. Default: None. Returns: Tensor, the dtype is same as input tensor. - If :attr:`keepdim` is True, the output shape is :math:`[N, 1]` or :math:`[1]`, depending on whether the input has data shaped as :math:`[N, D]`. - If :attr:`keepdim` is False, the output shape is :math:`[N]` or :math:`[]`, depending on whether the input has data shaped as :math:`[N, D]`. Examples: .. code-block:: python import paddle x = paddle.to_tensor([[1., 3.], [3., 5.]], dtype=paddle.float64) y = paddle.to_tensor([[5., 6.], [7., 8.]], dtype=paddle.float64) distance = paddle.nn.functional.pairwise_distance(x, y) print(distance.numpy()) # [5. 5.] porderZPairwiseDistanceepsilonkeepdimg)dtypeFaxisxZfloat32Zfloat64y)shapeZ fill_valuer)name)rr rr p_normXZOut)typeZinputsZoutputsattrs)rfloatintboolrrsubtractpaddleZfluidZdygraphbaseZ to_variableraddrr rZelementwise_subZelementwise_addrblockZ create_varfullrZ"create_variable_for_type_inferenceZ append_op) rrpr rrsubZ epsilon_varhelperroutr(MD:\Projects\ConvertPro\env\Lib\site-packages\paddle/nn/functional/distance.pypairwise_distancesX*       r*)r r FN) rZfluid.data_feederrrZfluid.layer_helperrrrZpaddle.fluid.frameworkrr __all__r*r(r(r(r)s