主要内容

生成的CNN类层次结构

生成的CNN代码具有以下类层次结构。图层类和生成的网络类有三个重要方法:

  1. 设置(),为每一层分配内存和系统资源。

  2. 预测(),在执行循环中执行推理。

  3. 清理(​​),将发布所有内存和系统资源。

相关话题