导入Python模块

导入昇腾AI软件栈提供的Python模块，代码示例如下所示。

import te.lang.cce
from te import tvm
from topi import generic

其中：

• “te.lang.cce”：引入TE支持的特定域语言接口，包括常见的运算vmuls、vadds、matmul等。

  具体的接口定义可查看DDK安装路径下“/site-packages/te-0.4.0.egg/te/lang/cce/”目录下的python函数，使用方法请参见《TE API参考》。

• “te.tvm”：引入TVM后端代码生成机制。

  具体的接口定义可查看DDK安装路径下“/site-packages/te-0.4.0.egg/te/tvm”目录下的python函数，使用方法请参见https://docs.tvm.ai/。

• “topi.generic”：提供了自动算子调度接口。

  具体的接口定义可查看DDK安装路径下“/site-packages/topi-0.4.0.egg/topi/generic”目录下的所有python函数，使用方法请参见《TE API参考》。