Files	Last commit	Last update
examples	slice_with_axes, slice_last_dim README Co-authored-by: zhoulong50_hw<zhoulong50@huawei.com> # message auto-generated for no-merge-commit merge: !4475 merge slice_docs into master slice_with_axes, slice_last_dim README Created-by: zhoulong50_hw Commit-by: zhoulong50_hw Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 补充算子 README 文档，并补充 geir 调用 examples ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2536 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新了 README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4475	3 months ago
op_graph	3个算子原型增加去重宏 Co-authored-by: zhoulong50_hw<zhoulong50@huawei.com> # message auto-generated for no-merge-commit merge: !4564 merge proto-add-dedup-macro into master 3个算子原型增加去重宏 Created-by: zhoulong50_hw Commit-by: zhoulong50_hw Merged-by: cann-robot Description: ## 描述为 DataCompare、SliceLastDim、SliceWithAxes 三个算子原型头文件增加去重宏保护（#ifndef OPS_PROTO_DEF_<OP> / #define / #endif），包裹 REG_OP ... OP_END_FACTORY_REG 块。在多仓组合编译或头文件交叉包含场景下，可避免同一算子被重复注册导致的编译/链接冲突。宏名按算子名大写、去分隔符命名：OPS_PROTO_DEF_DATACOMPARE、OPS_PROTO_DEF_SLICELASTDIM、OPS_PROTO_DEF_SLICEWITHAXES。本次仅增加预处理宏保护，不改变算子功能、数据类型、约束与调用路径。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2590 ## 测试算子级冒烟/泛化测试。 ## 文档更新无。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：算子原型去重宏保护（构建健壮性） See merge request: cann/ops-math!4564	2 months ago
op_host	Math仓算子适配Ascend350平台 Co-authored-by: huairuchen<chenhuairu@huawei.com> # message auto-generated for no-merge-commit merge: !6372 merge dvlite-math-350-adapt into master Math仓算子适配Ascend350平台 Created-by: huairuchen Commit-by: huairuchen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> Math仓部分随机数算子在350平台进行适配 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/4202 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 算子泛化测试，冒烟测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6372	13 days ago
op_kernel	slice_with_axes and slice_last_dim for ascend950 Co-authored-by: zhoulong50_hw<zhoulong50@huawei.com> # message auto-generated for no-merge-commit merge: !3315 merge sod-up into master slice_with_axes and slice_last_dim for ascend950 Created-by: zhoulong50_hw Commit-by: zhoulong50_hw Merged-by: cann-robot Description: ## 描述新增 SliceLastDim 和 SliceWithAxes 两个 conversion 类算子： - SliceLastDim：对输入张量的最后一个维度按 start/end/stride 进行切片，针对连续拷贝和跨步拷贝两种模式做了专门优化 - SliceWithAxes：对输入张量沿指定 axes 按 offsets/size 进行多轴并行切片包含完整的 proto、infershape、tiling、kernel 实现及 UT。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1964 ## 测试 - 二级冒烟通过 - ttk kernel 算子泛化精度测试：SliceLastDim 100/100 PASS，SliceWithAxes 100/100 PASS - infershape UT：各 5 条，全部 PASS - tiling UT：各 2 条，全部 PASS ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3315	4 months ago
tests	fix SliceWithAxes infershape dtype/-1/-2 Co-authored-by: wang-shilong32<wangshilong23@huawei.com> # message auto-generated for no-merge-commit merge: !6594 merge fix_swa_infershape into master fix SliceWithAxes infershape dtype/-1/-2 Created-by: wang-shilong32 Commit-by: wang-shilong32 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 SliceWithAxes 算子 InferShape 阶段对数据类型与未知维度处理不当的问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3817 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 冒烟、本地ut验证、出包测试用例。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6594	20 days ago
CMakeLists.txt	Math仓算子适配Ascend350平台 Co-authored-by: huairuchen<chenhuairu@huawei.com> # message auto-generated for no-merge-commit merge: !6372 merge dvlite-math-350-adapt into master Math仓算子适配Ascend350平台 Created-by: huairuchen Commit-by: huairuchen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> Math仓部分随机数算子在350平台进行适配 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/4202 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 算子泛化测试，冒烟测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6372	13 days ago
README.md	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	1 month ago
README_en.md	docs: add English translations for markdown files Co-authored-by: jenny_xiang<jenny.xiang@huawei.com> # message auto-generated for no-merge-commit merge: !7113 merge docs/add-english-translations into master docs: add English translations for markdown files Created-by: jenny_xiang Commit-by: jenny_xiang Merged-by: cann-robot Description: ## 描述新增仓库Markdown文档英文译稿（第二批，共409个文件）：第一批（已合入）之外的其余全部译稿，覆盖conversion、examples、ge、math、random各类算子README译稿，以及CHANGELOG_en.md、.gitcode PR模板英文版等。本批内全部仓内链接均指向master已有文件（含第一批已合入的21+2篇译稿），合入后本仓432篇英文文档全部就位，链接完整生效。本批同时包含4篇基于最新源文档的增量更新（math/ada_cast、math/prod_force_se_a、math/sinkhorn、math/xlogy，对应上游近期文档变更）。 ## 关联的Issue 无 ## 测试文档类变更，无需算子测试；已通过链接有效性校验（本批全部相对链接仅指向master已有文件，无指向本PR外未提交文件的链接），并完成逐篇翻译规范自查与语义校对（含支持状态表、限定词、术语、数值符号核对）。 ## 文档更新新增409个英文译稿文件（*_en.md / .en-US.md），不修改任何现有文件。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7113	6 days ago

SliceWithAxes

English | 简体中文

Product Support Status

Product	Supported
Ascend 950PR&950DT products	√
Atlas A3 products	√
Atlas A2 products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Function Description

Operator function: extracts slices from the input tensor along the specified axes, where offsets specifies the start position of each axis and size specifies the slice length of each axis.
Computation example:
- Input x shape = (10, 20, 30), axes = [0, 2], offsets = [1, 5], size = [5, 10]
- Output y shape = (5, 20, 10)
- y[i, j, k] = x[1 + i, j, 5 + k]

Computation formula: The shape of the output y is the same as that of the input x, except that the dimensions specified in axes are replaced with the corresponding sizes. For the k-th axis axis_k in axes:

y[..., axis_k, ...] = x[..., offsets[k] : offsets[k] + size[k], ...]

If size[k] == -1, the slice starts at offsets[k] and extends to the end of the axis, that is, size[k] = x.shape[axis_k] - offsets[k]

Parameter Description

Parameter Name	Input/Output/Attribute	Description	Data Type	Data Format
x	Input	Input tensor	INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT, FLOAT16, BF16, BOOL	ND
offsets	Input	Start position of the slice on each axis. This parameter is a 1D tensor whose length equals the length of axes	INT32, INT64	ND
size	Input	Slice length of each axis. This parameter is a 1D tensor whose length equals the length of axes. The value -1 indicates that the slice starts at the offset and extends to the end of the axis	INT32, INT64	ND
axes	Attribute (Required)	List of axes to be sliced	ListInt	-
y	Output	Output tensor, which has the same data type and format as x	Same as x	ND

Constraint Description

The dimension of the input tensor x ranges from [1, 8].
Each axis in axes must satisfy 0 <= axis < rank.
For each sliced axis, offsets[k] >= 0 and offsets[k] + size[k] <= x.shape[axis_k] must be satisfied (if size[k] is -1, the slice automatically extends to the end of the axis).
The size of each dimension of the output must be >= 0.
Dynamic shape, dynamic rank, and dynamic compilation with static shapes are supported.

Invocation Description

Invocation Method	Sample Code	Description
Graph-mode invocation	test_geir_slice_with_axes	Invokes the SliceWithAxes operator by building a graph through the operator IR.