ngraph.opset5.bucketize

ngraph.opset5.bucketize(data: _pyngraph.Node, buckets: Union[_pyngraph.Node, int, float, numpy.ndarray], output_type: str = 'i64', with_right_bound: bool = True, name: Optional[str] = None) _pyngraph.Node

Return a node which produces the Bucketize operation.

Parameters
  • data – Input data to bucketize

  • buckets – 1-D of sorted unique boundaries for buckets

  • output_type – Output tensor type, “i64” or “i32”, defaults to i64

  • with_right_bound – indicates whether bucket includes the right or left edge of interval. default true = includes right edge

  • name – Optional name for output node.

Returns

Bucketize node