昇腾社区首页
中文
注册
开发者
下载

aclnnForeachMaximumScalarV2

产品支持情况

[object Object]undefined

功能说明

  • 算子功能:

    返回一个和输入张量列表同样形状大小的新张量列表,对张量列表和标量值scalar执行逐元素比较,返回最大值的张量。

    本接口相较于,修改入参scalar的结构类型aclTensor为aclScalar,请根据实际情况选择合适的接口。

  • 计算公式:

    x=[x0,x1,...xn1]y=[y0,y1,...yn1]x = [{x_0}, {x_1}, ... {x_{n-1}}]\\ y = [{y_0}, {y_1}, ... {y_{n-1}}]\\ yi=max(xi,scalar)(i=0,1,...n1){\rm y}_i = \max(x_i, \text{scalar}) (i=0,1,...n-1)

函数原型

每个算子分为,必须先调用“aclnnForeachMaximumScalarV2GetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小,再调用“aclnnForeachMaximumScalarV2”接口执行计算。

  • [object Object]
  • [object Object]

aclnnForeachMaximumScalarV2GetWorkspaceSize

  • 参数说明

    • x(aclTensorList*,计算输入):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16和INT32。支持ND,shape维度不高于8维。不支持。该参数中所有Tensor的数据类型保持一致。
    • scalar(aclScalar*,计算输入):公式中的[object Object],Host侧的aclScalar,支持ND。不支持。数据类型支持FLOAT、FLOAT16、INT32、DOUBLE、INT64,且与入参[object Object]的数据类型具有一定对应关系,且数值不超过入参[object Object]对应数据类型的数值范围:
      • [object Object]的数据类型为FLOAT, BFLOAT16时,数据类型支持FLOAT, DOUBLE。
      • [object Object]的数据类型为FLOAT16时,数据类型支持FLOAT16, DOUBLE。
      • [object Object]的数据类型为INT32时,数据类型支持INT32, INT64。
    • out(aclTensorList*,计算输出):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16和INT32,支持ND,shape维度不高于8维。数据类型和数据格式与入参[object Object]的数据类型和数据格式一致,shape size大于等于入参[object Object]的shape size。不支持。该参数中所有Tensor的数据类型保持一致。
    • workspaceSize(uint64_t*,出参):返回用户需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor**,出参):返回op执行器,包含了算子计算流程。
  • 返回值

    aclnnStatus:返回状态码,具体参见

    [object Object]

aclnnForeachMaximumScalarV2

  • 参数说明

    • workspace(void*,入参):在Device侧申请的workspace内存地址。
    • workspaceSize(uint64_t,入参):在Device侧申请的workspace大小,由第一段接口aclnnForeachMaximumScalarV2GetWorkspaceSize获取。
    • executor(aclOpExecutor*,入参):op执行器,包含了算子计算流程。
    • stream(aclrtStream,入参):指定执行任务的Stream。
  • 返回值

    aclnnStatus:返回状态码,具体参见

约束说明

无。

调用示例

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

[object Object]