ops-math AsStrided算子文档

张开发
• 2026/5/9 18:13:55 • 15 分钟阅读

最新文章

推荐文章

相关文章

分享文章

ops-math AsStrided算子文档
AsStrided【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能允许用户通过制定新的形状size和步长stride来创建一个与原张量共享相同数据内存的张量视图。计算公式$$ out_iinput_{\text{storage_offset}\sum_{d0}^{D-1}(i_d\cdot \text{strided}[d])} $$参数说明参数名输入/输出/属性描述数据类型数据格式x输入公式中的input_i。INT64、UINT64、INT32、UINT32、FLOAT、FLOAT16、INT8、UINT8、BF16、INT16、UINT16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FNNDsize输入输出张量的形状。INT32、INT64NDstride输入stride[d]是输入张量在第d维的步幅。INT32、INT64NDstorage_offset输入是out_i中相对于原张量input_i存储的偏移量。INT32、INT64NDy输出公式中的out_i。INT64、UINT64、INT32、UINT32、FLOAT、FLOAT16、INT8、UINT8、BF16、INT16、UINT16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FNND约束说明无调用说明调用方式调用样例说明图模式调用test_geir_as_strided通过算子IR构图方式调用as_strided算子【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

更多文章