已合并
增加SparseReshape算子indices输入值的约束说明 #4136
kevin_huang1234创建于 22 天前
增加SparseReshape算子indices输入值的约束说明 #4136
已合并
共 1 个文件变更+1-0
| @@ -88,6 +88,7 @@ $$ | |||
| 88 | ## 约束说明 | 88 | ## 约束说明 |
| 89 | 89 | ||
| 90 | - 输入indices必须为2D矩阵,shape和new_shape必须为1D向量。 | 90 | - 输入indices必须为2D矩阵,shape和new_shape必须为1D向量。 |
| 91 | - 输入indices每行是一个非零元素的多维坐标。元素值大于等于0且小于对应shape维度的size。 | ||
| 91 | - input_rank和output_rank均不超过8维(MAX_RANK=8)。 | 92 | - input_rank和output_rank均不超过8维(MAX_RANK=8)。 |
| 92 | - new_shape中最多允许一个维度为-1,该维度值由总元素数除以其他维度乘积自动推导。 | 93 | - new_shape中最多允许一个维度为-1,该维度值由总元素数除以其他维度乘积自动推导。 |
| 93 | - 输入shape和输出new_shape的总元素数必须一致。 | 94 | - 输入shape和输出new_shape的总元素数必须一致。 |