onnx_helpers/nodes/ops/
reduce_mean.rs

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