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]]]