Class ov::op::v16::SegmentMax#

class SegmentMax : public ov::op::Op#

An operation which computes the maximum values along segments of a tensor.

Public Functions

SegmentMax(const Output<Node> &data, const Output<Node> &segment_ids, const op::FillMode fill_mode)#

Constructs a SegmentMax operation.

Parameters:

  • data – Input tensor with data

  • segment_ids – Indices of segments in the data input tensor

  • fill_mode – The value assigned to segments which are empty

SegmentMax(const Output<Node> &data, const Output<Node> &segment_ids, const Output<Node> &num_segments, const op::FillMode fill_mode)#

Constructs a SegmentMax operation.

Parameters:

  • data – Input tensor with data

  • segment_ids – Indices of segments in the data input tensor

  • num_segments – The segments count

  • fill_mode – The value assigned to segments which are empty

virtual void validate_and_infer_types() override#

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.