TensorRT:Conv + BN 融合

首先在複習一下BN公式:官方文檔  4個參數:均值,方差,gamma,beta   再在複習一下conv公式: 一般是conv以後接的是BN,所以conv輸入x,輸出y; BN的輸入就是y.  把y帶入BN的計算公式:  所以,會得到conv+bn融合以後的權重w和偏置bias <
相關文章
相關標籤/搜索