添加自定义layer有没有指导文档

模型有比较多不支持的layer,请问在SDK中说的添加自定义layer指导文档发布了吗

1 Like

您好,我们正在添加其他图层,请注明您需要的图层,@Frank 可能会为您提供帮助 :slight_smile:

这个目前还没有如何添加自定义层的相关文档。

2 Likes

@yuming 这个工具不是开源的,所以添加自定义的层是行不通的.这边建议是不支持的层做拆解,用支持的层做近似操作,或者简化模型

1 Like

ok,我需要的layer比较多,如upsample,插值,expand,permute等,会找替代方案,但也希望可以有自定义方式供参考,哪怕是跑cpu

1 Like

@yuming 目前来说没有,而且板子的CPU上面跑不是好选择,跟PC上的CPU差距是比较大的.只能寻找替换的方式,你可以逐个试试,具体是哪部分不支持,再做近似的考虑

2 Likes

我想再请教下,通道复制在现行支持layer下有没有好的实现方法,比如说,(1, 3, 512, 1)变成(1, 3, 512, 512),NCHW排列,方法是复制,在onnx里是expand。