ngraph.opset3.variadic_split¶
-
ngraph.opset3.
variadic_split
(data: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: Union[_pyngraph.Node, int, float, numpy.ndarray], split_lengths: Union[_pyngraph.Node, int, float, numpy.ndarray], name: Optional[str] = None) → _pyngraph.Node¶ Return a node which splits the input tensor into variadic length slices.
- Parameters
data – The input tensor to be split
axis – Axis along which the input data will be split
split_lengths – Sizes of the output tensors along the split axis
- Returns
VariadicSplit node