Merged
【unfold_grad】修复issue问题 aclnnUnfoldGrad文档添加对gradOut 末维必须等于 size约束的描述 #6716
tangkaidiAdd on 16 days ago
【unfold_grad】修复issue问题 aclnnUnfoldGrad文档添加对gradOut 末维必须等于 size约束的描述 #6716
Merged
Total: 1 files changed+2-1
| @@ -104,7 +104,7 @@ aclnnStatus aclnnUnfoldGrad( | |||
| 104 | <td>gradOut（aclTensor *）</td> | 104 | <td>gradOut（aclTensor *）</td> |
| 105 | <td>输入</td> | 105 | <td>输入</td> |
| 106 | <td>表示梯度更新系数。</td> | 106 | <td>表示梯度更新系数。</td> |
| 107 | - <td>shape为(..., (sizedim-size)/step+1, size)，要求满足gradOut的第dim维等于(inputSizes[dim]-size)/step+1和gradOut的size等于inputSizes的size+1。</td> | 107 | + <td>shape为(..., (sizedim-size)/step+1, size)，要求满足gradOut的第dim维等于(inputSizes[dim]-size)/step+1、gradOut的末维大小等于size和gradOut的size等于inputSizes的size+1。</td> |
| 108 | <td>FLOAT、FLOAT16、BFLOAT16</td> | 108 | <td>FLOAT、FLOAT16、BFLOAT16</td> |
| 109 | <td>ND</td> | 109 | <td>ND</td> |
| 110 | <td>1-8</td> | 110 | <td>1-8</td> |
| @@ -275,6 +275,7 @@ aclnnStatus aclnnUnfoldGrad( | |||
| 275 | 275 | ||
| 276 | 1. gradOut的shape满足约束： | 276 | 1. gradOut的shape满足约束： |
| 277 | - gradOut的第dim维等于(inputSizes[dim]-size)/step+1。 | 277 | - gradOut的第dim维等于(inputSizes[dim]-size)/step+1。 |
| 278 | + - gradOut的末维大小等于size。 | ||
| 278 | - gradOut的size等于inputSizes的size+1。 | 279 | - gradOut的size等于inputSizes的size+1。 |
| 279 | 2. dim、size、step的要求： | 280 | 2. dim、size、step的要求： |
| 280 | - size大于0且size小于等于inputSizes的第dim维。 | 281 | - size大于0且size小于等于inputSizes的第dim维。 |