mindspore.ops.pdist

mindspore.ops.pdist(input, p=2.0)

计算输入中每对行向量之间的p-范数距离。如果输入Tensor input 的shape为 $(N,M)$，那么输出就是一个shape为 $(N\ast (N-1)/2,)$ 的Tensor。

$$y[n]=\sqrt[p]{{\mid x_i-x_j\mid }^p}$$

其中 $x_i,x_j$ 是输入中两个不同的行向量。

参数：

• input (Tensor) - 输入Tensor。类型：float16、float32或float64。

• p (float) - 范数距离的阶， $p\in [0,\infty )$ 。默认值： 2.0 。

返回：

Tensor，类型与 input 一致。

异常：

• TypeError - input 不是Tensor。

• TypeError - input 的数据类型不是float16、float32、float64。

• TypeError - p 不是float。

• ValueError - p 是负数。

• ValueError - input 的维度小于2。

支持平台：

GPU CPU

样例：

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