关于pytorch的添加新的卷积层

大家好,本人刚接触pytorch不久,想要尝试新加一层自定义的卷积层,现在毫无头绪,也尝试读了源码,并没有发现卷积层的forward和backward示例,发现源码中卷积层的实现都是都是通过c扩展实现的,但是由于源码过于纷杂且本人学识有限,无法找到卷积层的具体实现(前向计算和反向计算)方法在哪里,望各位大佬相助
已邀请:

Cale

赞同来自:

您好~请问您解决了么?我也遇到同样的需求。通过C++API或者C扩展也只能使用内置扩展包,找不到Conv的源码。我在想Caffe是不是会好改一点?

要回复问题请先登录注册