Deep learning framework by BAIR
Created by Yangqing Jia Lead Developer Evan Shelhamer
Reduction
./include/caffe/layers/reduction_layer.hpp
./src/caffe/layers/reduction_layer.cpp
./src/caffe/layers/reduction_layer.cu
ReductionParameter reduction_param
./src/caffe/proto/caffe.proto
// Message that stores parameters used by ReductionLayer message ReductionParameter { enum ReductionOp { SUM = 1; ASUM = 2; SUMSQ = 3; MEAN = 4; } optional ReductionOp operation = 1 [default = SUM]; // reduction operation // The first axis to reduce to a scalar -- may be negative to index from the // end (e.g., -1 for the last axis). // (Currently, only reduction along ALL "tail" axes is supported; reduction // of axis M through N, where N < num_axes - 1, is unsupported.) // Suppose we have an n-axis bottom Blob with shape: // (d0, d1, d2, ..., d(m-1), dm, d(m+1), ..., d(n-1)). // If axis == m, the output Blob will have shape // (d0, d1, d2, ..., d(m-1)), // and the ReductionOp operation is performed (d0 * d1 * d2 * ... * d(m-1)) // times, each including (dm * d(m+1) * ... * d(n-1)) individual data. // If axis == 0 (the default), the output Blob always has the empty shape // (count 1), performing reduction across the entire input -- // often useful for creating new loss functions. optional int32 axis = 2 [default = 0]; optional float coeff = 3 [default = 1.0]; // coefficient for output }