ConfusionTransposeD
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR/Ascend 950DT |
√ |
功能说明
- 算子功能:融合reshape和transpose运算。
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| x |
输入 |
输入张量。 |
INT8、INT16、 INT32、 INT64、UINT8、UINT16、UINT32、UINT64、FLOAT16、FLOAT、BFLOAT16 |
ND |
| perm |
输入 |
转置后每根轴对应的转置前轴索引。 |
INT64 |
ND |
| shape |
输入 |
reshape后的shape大小。 |
INT64 |
ND |
| transposeFirst |
输入 |
判断是否先执行transpose操作。 |
BOOL |
ND |
| y |
输出 |
输出张量。 |
与输入x保持一致。 |
ND |
约束说明
-perm张量中元素必须唯一,并在[0,perm的维度数量-1]范围内。
-当transposeFirst为True时,perm的长度必须与x的shape的长度相同,即len(perm)=len(x_shape);当transposeFirst为False时,perm长度必须与属性输入shape的长度相同,即len(perm)=len(shape)。
-shape中的所有维度乘积必须等于输入张量x的元素总数。
调用说明