majianjia / nnom

A higher-level Neural Network library for microcontrollers.
Apache License 2.0
908 stars 245 forks source link

layer有多个输出,并且给不同next-layer场景,当前框架能否支持? #179

Open wuhuangcangg opened 1 year ago

wuhuangcangg commented 1 year ago

从源码看,tensor实际上绑定在layer中。 后续可能会补充算子,如slice,他有多个输出,每个输出都给到不同的layer。 多个输出场景,假如layer0有输出 output1, output2, 分别给到 layer1,layer2, 目前的框架在hook的时候,layer1、layer2似乎不能区分 layer0的哪个output是给它们的?

majianjia commented 1 year ago

是的,现有框架不能区分slice之后的两块数据。 如果要添加slice之类的多输出,需要修改框架,新增一个hook的接口指定第几个输出tensor