Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
wuhuangcangg opened this issue Feb 8, 2023 · 1 comment

Comments

@wuhuangcangg
Copy link

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

@majianjia
Copy link
Owner

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants