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 - inputother 不是Tensor。

  • TypeError - inputother 数据类型不是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