openvino.runtime.opset11.variadic_split

openvino.runtime.opset11.variadic_split(data: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], axis: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], split_lengths: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], name: Optional[str] = None) openvino._pyopenvino.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