onnx_helpers/nodes/ops/
concat.rs

1//! Concat operation.
2
3use crate::{builder, node_to_inner, nodes::Node};
4
5/// Concat node.
6pub struct Concat {
7    inner: Node,
8}
9
10impl Concat {
11    /// Creates new Concat operation.
12    #[inline(always)]
13    pub fn new<I>(axis: i64, inputs: I) -> Self
14    where
15        I: IntoIterator,
16        I::Item: Into<String>,
17    {
18        Concat {
19            inner: builder::Node::new("Concat")
20                .inputs(inputs)
21                .attribute("axis", axis)
22                .build(),
23        }
24    }
25}
26
27node_to_inner!(Concat);