mnn, python

1. mnn的卷积计算

mnn 卷积计算的输入tensor采用NC4HW4的数据格式,卷积计算的结果也是NC4HW4格式,所以一个通道,一个通道,计算4个元素。这里debug前4个元素进行对比。
在这里插入图片描述

2. python 计算结果

python采用同样的输入,weight,bias,然后debug前4个通道的数据
在这里插入图片描述
可以看到前4个结果是一样的

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐