ncnn icon indicating copy to clipboard operation
ncnn copied to clipboard
verified publisher icon Tencent

unfold+matmul的操作在ncnn中是否有好的替代方案？

Open EdVince opened this issue 4 years ago • 1 comments

detail | 详细描述 | 詳細な説明

如下述代码： unfold_a = F.unfold(input_a, kernel_size=4, stride=4).permute(0, 2, 1) unfold_b = F.unfold(input_b, kernel_size=4, stride=4) output = torch.matmul(unfold_a, unfold_b)

EdVince avatar Dec 05 '21 14:12 EdVince

ncnn已经支持 unfold 和 matmul，更新新版本试试

nihui avatar Apr 30 '25 02:04 nihui