mindspore.ops.Pdist
- class mindspore.ops.Pdist(p=2.0)[源代码]
计算输入中每对行向量之间的p-范数距离。
更多细节请参考
mindspore.ops.pdist()
。- 参数:
p (float,可选) - 范数距离的阶, \(p∈[0, ∞)\)。默认值:2.0。
- 输入:
x (Tensor) - 输入Tensor x ,其shape为 \((*B, N, M)\),其中 \(*B\) 表示批处理大小,可以是多维度。类型: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]