TensorRT:Conv + BN 融合

首先在复习一下BN公式:官方文档  4个参数:均值,方差,gamma,beta   再在复习一下conv公式: 一般是conv以后接的是BN,所以conv输入x,输出y; BN的输入就是y.  把y带入BN的计算公式:  所以,会得到conv+bn融合以后的权重w和偏置bias <
相关文章
相关标签/搜索