mindspore.ops.Cdist

class mindspore.ops.Cdist(p=2.0)

计算两个Tensor的p-norm距离。

更多参考详见 mindspore.ops.cdist()。

参数：

• p (float，可选) - 计算向量对p-norm距离的P值，P∈[0，∞]。默认值： 2.0 。

输入：

• input_x (Tensor) - 输入Tensor，shape为 \((B, P, M)\) ， \(B\) 维度为0时该维度被忽略，shape为 \((P, M)\) 。

• input_y (Tensor) - 输入Tensor，shape为 \((B, R, M)\) ，与 input_x 的数据类型一致。

输出：

Tensor，p-norm距离，数据类型与 input_x 一致，shape为 \((B, P, R)\)。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([[[1.0, 1.0], [2.0, 2.0]]]).astype(np.float32))
>>> input_y = Tensor(np.array([[[3.0, 3.0], [3.0, 3.0]]]).astype(np.float32))
>>> op = ops.Cdist(p=2.0)
>>> output = op(input_x, input_y)
>>> print(output)
[[[2.8284273 2.8284273]
  [1.4142137 1.4142137]]]