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

aclnnAll

产品支持情况

[object Object]undefined

功能说明

  • 接口功能:对于给定维度[object Object]中的每一维,如果输入Tensor中该维度对应的所有元素计算为True,则返回True,否则返回False。如果[object Object]为True,则输出Tensor的维度与输入相同,否则,[object Object]维将会被压缩,导致输出Tensor减少[object Object]个维度。
  • 例如,输入Tensor的shape是:(A×B×C×D)(A\times B \times C \times D)[object Object]值为[0, 2],如果[object Object]为False,则输出Tensor的shape是:(B×D)(B \times D),输出Tensor比输入少两维,如果[object Object]为True,则输出Tensor的shape是:(1×B×1×D)(1 \times B \times 1 \times D),输出Tensor的维度与输入相同。

函数原型

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

[object Object]
[object Object]

aclnnAllGetWorkspaceSize

  • 参数说明:

    [object Object]
    • [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas 200I/500 A2 推理产品[object Object]:不支持BFLOAT16数据类型。
  • 返回值:

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

    第一段接口完成入参校验,出现以下场景时报错:

    [object Object]

aclnnAll

  • 参数说明:

    [object Object]
  • 返回值:

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

约束说明

  • 确定性计算:
    • aclnnAll默认确定性实现。

调用示例

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

[object Object]