openvino.runtime.opset6.split

openvino.runtime.opset6.split(data: Node | int | float | ndarray, axis: Node | int | float | ndarray, num_splits: int, name: str | None = None) → 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