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

ConvolutionDepthWise不支持带有d维数据的卷积吗 #5661

Open
zhang0557kui opened this issue Aug 31, 2024 · 2 comments
Open

ConvolutionDepthWise不支持带有d维数据的卷积吗 #5661

zhang0557kui opened this issue Aug 31, 2024 · 2 comments

Comments

@zhang0557kui
Copy link

pytorch模型通过pnnx转到ncnn,运行发现ncnn的ConvolutionDepthWise算子在算卷积时,把我的d维数据丢了,输出d=1,是目前还不支持[d, c, h, w]的数据卷积吗

@wzyforgit
Copy link
Contributor

3d转换是支持的,估计是个BUG,提供一下pytorch的最小复现代码?

@zhang0557kui
Copy link
Author

3d转换是支持的,估计是个BUG,提供一下pytorch的最小复现代码?

spatial_layer = nn.Conv2d(256, 256, 3, 1, 0, groups=256, bias=False)
model = spatial_layer.eval()
x = torch.randn(98, 256, 3, 3)
print(model(x).shape)

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