标签： tensorflow pytorch batch-normalization
我正在尝试将旧的tensorflow / keras网络转换为pytorch，我对我获得的batch_normalization（BN）权重的值感到困惑。例如，我的火炬网络的第一BN层具有两组形状为[8]的元素（权重和偏差）。相反，张量流中的同一BN层返回4组形状为[8]的元素。
由于我不打算在将网络转换为pytorch后再训练网络，所以我想知道tensorflow BN层的哪两组元素与pytorch BN层相对应。