| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[feat] 新增embedding算子 Co-authored-by: warming<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !2000 merge master into master [feat] 新增embedding算子 Created-by: wsunmoon Commit-by: warming Merged-by: cann-robot Description: ## 描述 950新增embedding算子，包含算子原型、tiling、kernel、SIMT模板等。该算子把数据集合映射到向量空间，进而将数据进行量化。 例如：输入二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下： $$ \begin{aligned} & weight = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{11} & x_{12} & ... & x_{1n}\\ x_{21} & x_{22} & ... & x_{2n}\\ ...\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ & indices = \begin{bmatrix} 0,2,m \end{bmatrix}\\ & out = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{21} & x_{22} & ... & x_{2n}\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ \end{aligned} $$ ## 关联的Issue 关联Issue [#1360](https://gitcode.com/cann/ops-nn/issues/1360) ## 测试 UT，ST，线上冒烟均已通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2000 | 3 个月前 | |
[feat] 新增embedding算子 Co-authored-by: warming<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !2000 merge master into master [feat] 新增embedding算子 Created-by: wsunmoon Commit-by: warming Merged-by: cann-robot Description: ## 描述 950新增embedding算子，包含算子原型、tiling、kernel、SIMT模板等。该算子把数据集合映射到向量空间，进而将数据进行量化。 例如：输入二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下： $$ \begin{aligned} & weight = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{11} & x_{12} & ... & x_{1n}\\ x_{21} & x_{22} & ... & x_{2n}\\ ...\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ & indices = \begin{bmatrix} 0,2,m \end{bmatrix}\\ & out = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{21} & x_{22} & ... & x_{2n}\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ \end{aligned} $$ ## 关联的Issue 关联Issue [#1360](https://gitcode.com/cann/ops-nn/issues/1360) ## 测试 UT，ST，线上冒烟均已通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2000 | 3 个月前 | |
[feat] 新增embedding算子 Co-authored-by: warming<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !2000 merge master into master [feat] 新增embedding算子 Created-by: wsunmoon Commit-by: warming Merged-by: cann-robot Description: ## 描述 950新增embedding算子，包含算子原型、tiling、kernel、SIMT模板等。该算子把数据集合映射到向量空间，进而将数据进行量化。 例如：输入二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下： $$ \begin{aligned} & weight = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{11} & x_{12} & ... & x_{1n}\\ x_{21} & x_{22} & ... & x_{2n}\\ ...\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ & indices = \begin{bmatrix} 0,2,m \end{bmatrix}\\ & out = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{21} & x_{22} & ... & x_{2n}\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ \end{aligned} $$ ## 关联的Issue 关联Issue [#1360](https://gitcode.com/cann/ops-nn/issues/1360) ## 测试 UT，ST，线上冒烟均已通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2000 | 3 个月前 | |
[feat] 新增embedding算子 Co-authored-by: warming<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !2000 merge master into master [feat] 新增embedding算子 Created-by: wsunmoon Commit-by: warming Merged-by: cann-robot Description: ## 描述 950新增embedding算子，包含算子原型、tiling、kernel、SIMT模板等。该算子把数据集合映射到向量空间，进而将数据进行量化。 例如：输入二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下： $$ \begin{aligned} & weight = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{11} & x_{12} & ... & x_{1n}\\ x_{21} & x_{22} & ... & x_{2n}\\ ...\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ & indices = \begin{bmatrix} 0,2,m \end{bmatrix}\\ & out = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{21} & x_{22} & ... & x_{2n}\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ \end{aligned} $$ ## 关联的Issue 关联Issue [#1360](https://gitcode.com/cann/ops-nn/issues/1360) ## 测试 UT，ST，线上冒烟均已通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2000 | 3 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 3 个月前 | ||
| 3 个月前 | ||
| 3 个月前 | ||
| 3 个月前 |