layer有多个输出，并且给不同next-layer场景，当前框架能否支持？
wuhuangcangg opened this issue · 1 comments
wuhuangcangg commented
从源码看，tensor实际上绑定在layer中。
后续可能会补充算子，如slice，他有多个输出，每个输出都给到不同的layer。
多个输出场景，假如layer0有输出 output1， output2, 分别给到 layer1，layer2，
目前的框架在hook的时候，layer1、layer2似乎不能区分 layer0的哪个output是给它们的？
majianjia commented
是的，现有框架不能区分slice之后的两块数据。
如果要添加slice之类的多输出，需要修改框架，新增一个hook的接口指定第几个输出tensor