Edit this page

ngraph.split

ngraph.split(data: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: Union[_pyngraph.Node, int, float, numpy.ndarray], num_splits: int, name: Optional[str] = None) → _pyngraph.Node

Return a node which splits the input tensor into same-length slices.

Parameters

• data – The input tensor to be split

• axis – Axis along which the input data will be split

• num_splits – Number of the output tensors that should be produced

Returns

Split node

Prev Next