Dot

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 接口功能:计算两个一维张量的点积结果。

  • 计算公式:

self=[x1,x2,...,xn]self = [x_{1}, x_{2}, ..., x_{n}]

tensor=[y1,y2,...,yn]tensor = [y_{1}, y_{2}, ..., y_{n}]

out=x1∗y1+x2∗y2+...+xn∗ynout = x_{1}*y_{1} + x_{2}*y_{2} + ... + x_{n}*y_{n}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入 参与点积计算的输入张量,公式中的self。与tensor、out的数据类型一致,与tensor的shape一致。 BFLOAT16、FLOAT16、FLOAT、INT8、INT32、UINT8 ND
tensor 输入 参与点积计算的输入张量,公式中的tensor。与self、out的数据类型一致,与self的shape一致。 BFLOAT16、FLOAT16、FLOAT、INT8、INT32、UINT8 ND
out 输出 点积计算结果,公式中的输出out。与self、tensor的数据类型一致。 BFLOAT16、FLOAT16、FLOAT、INT8、INT32、UINT8 ND
  • Atlas 推理系列产品、Atlas 训练系列产品:数据类型不支持BFLOAT16。

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_dot 通过aclnnAbs接口方式调用Dot算子。