CANN/pypto hypot函数

pypto.hypot

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

逐元素计算 input 和 other 的平方和的平方根（即直角三角形的斜边长）。计算公式如下：

$$ res_i = \sqrt{input_i^2 + other_i^2} $$

函数原型

hypot(input: Tensor, other: Tensor) -> Tensor

参数说明

返回值说明

返回输出Tensor，Tensor的数据类型和input、other相同，Shape为input和other广播后大小。

约束说明

1. input 和 other 类型应该相同。
2. other 不支持nan、inf等特殊值。
3. 对于 BF16 和 FP16 类型，内部计算可能会提升精度以避免中间溢出。

调用示例

TileShape设置示例

说明：调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输出一致。

示例1：输入input shape为[m, n]，输出为[m, n], TileShape设置为[m1, n1], 则m1, n1分别用于切分m, n轴。

pypto.set_vec_tile_shapes(4, 16)

接口调用示例

# 示例：计算两组直角边的斜边 # 第一组: (3, 4) -> 5 # 第二组: (5, 12) -> 13 a = pypto.tensor([3.0, 5.0], pypto.DT_FP32) b = pypto.tensor([4.0, 12.0], pypto.DT_FP32) out = pypto.hypot(a, b)

结果示例如下：

输入数据a: [3.0, 5.0] 输入数据b: [4.0, 12.0] 输出数据out: [5.0, 13.0]

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto