mindspore.ops.dist
- mindspore.ops.dist(input, other, p=2)[源代码]
计算输入中每对行向量之间的 \(p\)-norm距离。
说明
在MindSpore中只支持计算整数 \(p\)-norm形式的范数,如果 \(p\) 不是整数会引发类型错误。
- 参数:
input (Tensor) - 第一个输入Tensor,数据类型需为float16或float32。
other (Tensor) - 第二个输入Tensor,数据类型需为float16或float32。
p (int,可选) - 范数的次数。 p 大于或等于0。默认值:
2
。
- 返回:
Tensor,具有与 input 相同的dtype,其shape为:\((1)\) 。
- 异常:
TypeError - input 或 other 不是Tensor。
TypeError - input 和 other 数据类型不是float16或float32。
TypeError - p 不是非负整数。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> from mindspore import Tensor, ops >>> input_x = Tensor([[[1.0, 1.0], [2.0, 2.0]]]) >>> input_y = Tensor([[[3.0, 3.0], [3.0, 3.0]]]) >>> out = ops.dist(input_x, input_y) >>> print(out.asnumpy()) 3.1622777