Crate caffe2op_channelstats

source ·

Structs

  • ChannelStatsOp
    | Given an input tensor in NCHW format, | computes the sum of all elements per | channel and the sum of all elements squared | per channel. These values can be reduced | across multiple batches and used to | obtain the mean and variance across | the full set of batches. Using the new | mean and variance as input to SpatialBN | has the effect of changing the batch | size over which SpatialBN is applied. |