mindspore.ops.Pdist

class mindspore.ops.Pdist(p=2.0)[源代码]

计算输入中每对行向量之间的p-范数距离。

更多细节请参考 mindspore.ops.pdist()

说明

Pdist算子由于涉及求幂次方运算,在float16类型输入场景下,很容易会因为数值溢出而产生inf/nan的计算结果,建议采用float32类型输入。

参数:
  • p (float,可选) - 范数距离的阶, \(p∈[0, ∞)\)。默认值: 2.0

输入:
  • x (Tensor) - 输入Tensor。类型:float16、float32或float64。

输出:

Tensor,类型与 x 一致。

支持平台:

GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> import numpy as np
>>> x = Tensor(np.array([[1.0, 1.0], [2.0, 2.0], [3.0, 3.0]]).astype(np.float32))
>>> op = ops.Pdist(p=2.0)
>>> y = op(x)
>>> print(y)
[1.4142135 2.828427  1.4142135]