caffe2op-accum

Crates.iocaffe2op-accum
lib.rscaffe2op-accum
version0.1.5-alpha.0
sourcesrc
created_at2023-03-02 02:50:10.216853
updated_at2023-03-25 12:03:14.257517
descriptionxxx
homepage
repositoryhttps://github.com/kleb6/caffe2-rs
max_upload_size
id798573
size75,100
(klebs6)

documentation

https://docs.rs/caffe2op-accum

README

The caffe2op-accum crate provides an operator for accumulating values in networked digital signal processing and deep learning computations. This crate defines the AccumulateOp which performs accumulation on the input tensors according to a specified axis. The resulting tensor is returned as the output. The Accumulated token is used to refer to the output tensor.

Note: This crate is currently being translated from C++ to Rust, and some function bodies may still be in the process of translation.

The operation performed by AccumulateOp can be represented mathematically as follows:

If the input tensor is of shape [a, b, c, ..., m, n] and the accumulation axis is k, then the output tensor will be of shape [a, b, c, ..., l, n] where l is the size of the tensor along the accumulation axis. The elements of the output tensor are calculated as follows:

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

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

where the sum is over all l along the accumulation axis.

The caffe2op-accum crate also provides tokens such as Accumulation, accumulates, accumulations, depends, fiddles, interim, and reshaped, which are used within the implementation of the accumulation operation.

Commit count: 105

cargo fmt