Merged
【chunk_cat】修复issue问题 资料内功能说明处中添加对输出shape的说明 #6736
tangkaidiAdd on 18 days ago
【chunk_cat】修复issue问题 资料内功能说明处中添加对输出shape的说明 #6736
Merged
Total: 1 files changed+14-2
| @@ -23,7 +23,16 @@ | |||
| 23 | 23 | ||
| 24 | ## 功能说明 | 24 | ## 功能说明 |
| 25 | 25 | ||
| 26 | -将tensors中所有tensor先按照维度dim切分为numChunks块，再按照dim后一维进行级联，最后转换为out的数据类型。 | 26 | +当前仅支持dim为0。将tensors中的每个非空tensor沿首维划分为numChunks个等长块，每块的首维大小为ceil(原首维大小/numChunks)，其中ceil表示向上取整。若首维大小不能被numChunks整除，则在首维末尾补0。将各块展平为一维后，按输入列表顺序拼接各输入中序号相同的块，得到二维输出，最后转换为out的数据类型。 |
| 27 | + | ||
| 28 | +例如，两个输入tensor的shape均为[4, 3, 2]，numChunks为2时，每个输入的每块包含12个元素，拼接后每行包含24个元素，因此输出shape为[2, 24]。 | ||
| 29 | + | ||
| 30 | +首维大小不需要被numChunks整除。例如，输入列表仅包含一个shape为[5]、数据为[1, 2, 3, 4, 5]的一维tensor，numChunks为2时，每块长度为3。末尾补一个0后分成两块，输出shape为[2, 3]，数据为： | ||
| 31 | + | ||
| 32 | +```text | ||
| 33 | +[[1, 2, 3], | ||
| 34 | + [4, 5, 0]] | ||
| 35 | +``` | ||
| 27 | 36 | ||
| 28 | ## 函数原型 | 37 | ## 函数原型 |
| 29 | 38 | ||
| @@ -106,7 +115,7 @@ aclnnStatus aclnnChunkCat( | |||
| 106 | <tr> | 115 | <tr> |
| 107 | <td>out（aclTensor*）</td> | 116 | <td>out（aclTensor*）</td> |
| 108 | <td>输出</td> | 117 | <td>输出</td> |
| 109 | - <td>输出tensor。</td> | 118 | + <td>输出tensor，shape为功能说明中的二维输出shape。</td> |
| 110 | <td>数据类型所占字节数需要是输入数据类型所占字节数的非0整数倍。</td> | 119 | <td>数据类型所占字节数需要是输入数据类型所占字节数的非0整数倍。</td> |
| 111 | <td>FLOAT、FLOAT16、BFLOAT16</td> | 120 | <td>FLOAT、FLOAT16、BFLOAT16</td> |
| 112 | <td>ND</td> | 121 | <td>ND</td> |
| @@ -175,6 +184,9 @@ aclnnStatus aclnnChunkCat( | |||
| 175 | <tr> | 184 | <tr> |
| 176 | <td>numChunks小于等于0。</td> | 185 | <td>numChunks小于等于0。</td> |
| 177 | </tr> | 186 | </tr> |
| 187 | + <tr> | ||
| 188 | + <td>out的shape与期望输出shape不一致。</td> | ||
| 189 | + </tr> | ||
| 178 | </tbody> | 190 | </tbody> |
| 179 | </table> | 191 | </table> |
| 180 | 192 | ||