昇腾社区首页
中文
注册

算子模式使用流程

单算子使用流程

单算子使用的流程图如图1所示。

图1 单算子使用流程

具体使用方式以及样例请参见单算子使用章节。

图使用流程

图使用的本质为组合多个算子,形成一个图算子,以简化重复多算子的调用步骤,图使用的流程图如图2所示。

用户需要自行设计并定义图的结构,即Graph算子中Operation的组织,以及图输入,输出和中间Tensor。

  • 图输入Tensor为整个图的输入Tensor,即用户使用图算子时需要从外部输入的所有Tensor。
  • 图输出Tensor为整个图的输出Tensor,即用户使用图算子时不会再进行下一步运算操作的所有Tensor。
  • 图中间Tensor为图算子运算中产生的临时Tensor。
图2 图调用流程

具体使用方式以及样例请参见图使用章节。