aclnnPdistForward

支持的产品型号

• Atlas 训练系列产品。
• Atlas A2训练系列产品。

接口原型

每个算子分为，必须先调用“aclnnPdistForwardGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnPdistForward”接口执行计算。

• aclnnStatus aclnnPdistForwardGetWorkspaceSize(const aclTensor *self, const aclScalar *p, aclTensor *Out, uint64_t *workspaceSize, aclOpExecutor **executor)
• aclnnStatus aclnnPdistForward(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

功能描述

• 算子功能：计算输入中每对行向量的p范数距离。
• 计算公式： 设输入self的shape为M * N，$self_{in}$表示self第i行下标为n的元素，第i行和第j行的范数距离公式如下：$$out_{ij} = \sum_{k=0}^{N-1}||self_{in}-self_{jn}||_p = (\sum_{k=0}^{N-1}|self_{in}-self_{jn}|^p)^\frac{1}{p}$$ out的shape为M(M-1)/2，其中元素的排列顺序为$out_{01}$, ... ,$out_{0M}$,$out_{12}$, ... ,$out_{M-1,M}$

aclnnPdistForwardGetWorkspaceSize

• 参数说明：

  • self（aclTensor*, 计算输入）：Device侧的aclTensor，公式中的输入self，数据类型支持 FLOAT、FLOAT16，shape必须为二维，与out的数据类型需满足数据类型推导规则（参见），支持，支持ND。

  • p（aclScalar*，计算输入）：Device侧的aclScalar，公式中的输入p，表示范数的系数，支持数据类型为FLOAT64，其中的数据范围为[0, +∞]。

  • out（aclTensor*, 计算输出）：Device侧的aclTensor，范数的计算结果，数据类型支持 FLOAT、FLOAT16，与self的数据类型需满足数据类型推导规则（参见），支持，支持ND。

  • workspaceSize（uint64_t*, 出参）：返回用户需要在Device侧申请的workspace大小。

  • executor（aclOpExecutor**, 出参）：返回op执行器，包含了算子计算流程。

• 返回值：

  aclnnStatus： 返回状态码，具体参见。

  [object Object]

aclnnPdistForward

• 参数说明：

  • workspace（void*, 入参）：在Device侧申请的workspace内存地址。

  • workspaceSize（uint64_t, 入参）：在Device侧申请的workspace大小，由第一段接口aclnnPdistForwardGetWorkspaceSize获取。

  • stream（aclOpExecutor*, 入参）：指定执行任务的AscendCL Stream流。

  • executor（aclrtStream, 入参）：op执行器，包含了算子计算流程。

• 返回值：

  aclnnStatus： 返回状态码，具体参见。

约束与限制

无。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。