Lib.rs

caffe2op-accum crate 提供了一个在网络的数字信号处理和深度学习计算中累积值的操作符。这个 crate 定义了 AccumulateOp，它根据指定的轴对输入张量进行累积。结果张量作为输出返回。使用 Accumulated 标记来引用输出张量。

注意：该 crate 正在从 C++ 转译为 Rust，某些函数体可能仍在翻译过程中。

AccumulateOp 执行的操作可以用以下方式表示

如果输入张量的形状为 [a, b, c, ..., m, n] 并且累积轴为 k，则输出张量的形状将为 [a, b, c, ..., l, n] 其中 l 是累积轴上的张量大小。输出张量的元素按以下方式计算

output[i1, i2, ..., il, j] 

= sum(input[i1, i2, ..., i_{k-1}, l, i_{k+1}, ..., i_{m}, j])

其中求和是对累积轴上的所有 l 进行的。

caffe2op-accum crate 还提供了一些标记，如 Accumulation、accumulates、accumulations、depends、fiddles、interim 和 reshaped，这些标记在累积操作的实现中使用。