aclnnAny
产品支持情况
功能说明
对于给定维度[object Object]中的每一维,如果输入Tensor中该维度对应的任意元素计算为True,则返回True,否则返回False。如果[object Object]为True,则输出Tensor的维度与输入相同,否则,[object Object]维将会被压缩,导致输出Tensor减少[object Object]个维度。
例如:输入Tensor的shape是,[object Object]值为[0, 2],则输出Tensor的shape是,输出Tensor比输入少两维。
函数原型
每个算子分为,必须先调用“aclnnAnyGetWorkspaceSize”接口获取入参并根据流程计算所需workspace大小,再调用“aclnnAny”接口执行计算。
[object Object]
[object Object]
aclnnAnyGetWorkspaceSize
参数说明:
[object Object]- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas 200I/500 A2 推理产品[object Object]:不支持BFLOAT16数据类型。
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
aclnnAny
约束说明
- 确定性计算:
- aclnnAny默认确定性实现。
调用示例
[object Object]