Crate caffe2op_accum

source ·

Structs

  • | Accumulate operator accumulates the | input tensor to the output tensor. If | the output tensor already has the right | size, we add to it; otherwise, we first | initialize the output tensor to all | zeros, and then do accumulation. Any | further calls to the operator, given | that no one else fiddles with the output | in the interim, will do simple accumulations. | | Accumulation is done using Axpby operation | as shown: | | Y = 1X + gammaY | | where X is the input tensor, Y is the output | tensor and gamma is the multiplier argument. |