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

样例：

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