问题描述:
【功能模块】
C++拓展
【操作步骤&问题现象】
模型开发过程中遇到需要自定义Tensor运算逻辑的情况。Pytorch可以通过torch.utils.cpp_extension扩展C++代码,请问mindspore是否有类似功能支持?
解答:
可以的,看一下这两个教程 基于自定义算子接口调用第三方算子库 — MindSpore master documentation
自定义算子(基于Custom表达) — MindSpore master documentation

问题描述:
【功能模块】
C++拓展
【操作步骤&问题现象】
模型开发过程中遇到需要自定义Tensor运算逻辑的情况。Pytorch可以通过torch.utils.cpp_extension扩展C++代码,请问mindspore是否有类似功能支持?
解答:
可以的,看一下这两个教程 基于自定义算子接口调用第三方算子库 — MindSpore master documentation
自定义算子(基于Custom表达) — MindSpore master documentation