MrgbaCustom
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | × |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | × |
功能说明
-
算子功能：完成张量rgb和张量alpha的透明度乘法计算。
-
计算公式：
out=rgb∗alpha255out = rgb * \frac{alpha}{255}
其中，alpha会广播到与rgb相同的shape。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| rgb | 输入 | Device侧的aclTensor，shape支持HWC(C=3)，与alpha满足broadcast关系，只支持连续Tensor。 | UINT8 | ND |
| alpha | 输入 | Device侧的aclTensor，shape支持HWC(C=1)，与rgb满足broadcast关系，只支持连续Tensor。 | UINT8 | ND |
| out | 输出 | Device侧的aclTensor，shape支持HWC(C=3)，与rgb的shape一致，只支持连续Tensor。 | UINT8 | ND |
约束说明
- 输入shape限制：rgb的shape为HWC(C=3)，alpha的shape为HWC(C=1)，out的shape与rgb一致，且与alpha满足broadcast关系。
- 确定性计算：aclnnMrgbaCustom默认确定性实现。
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_aclnn_mrgba_custom | 通过aclnnMrgbaCustom接口方式调用MrgbaCustom算子。 |