onnx_helpers/nodes/ops/
reduce_sum.rs

1//! Reduce sum operation.
2
3use onnx_pb::Axes;
4
5use crate::{builder, node_to_inner, nodes::Node};
6
7/// Reduce sum node.
8pub struct ReduceSum {
9    inner: Node,
10}
11
12impl ReduceSum {
13    /// Creates new reduce sum operation.
14    #[inline(always)]
15    pub fn new<S: Into<String>, A: Into<Axes>>(input: S, axes: A, keepdims: bool) -> Self {
16        ReduceSum {
17            inner: builder::Node::new("ReduceSum")
18                .input(input)
19                .attribute("axes", axes.into())
20                .attribute("keepdims", keepdims)
21                .build(),
22        }
23    }
24}
25
26node_to_inner!(ReduceSum);