struct ngraph::runtime::reference::UniqueElements

Overview

#include <unique.hpp>

template <typename Index_t, typename Count_t>
struct UniqueElements
{
    // fields

    std::vector<TensorSlice<Index_t, Count_t>> all_tensor_elements;
    std::vector<TensorSlice<Index_t, Count_t>> unique_tensor_elements;
    int64_t axis = 0;
};

Detailed Documentation

Fields

std::vector<TensorSlice<Index_t, Count_t>> all_tensor_elements

Contains descriptors of all elements in the input tensor. Possibly sorted by value.

std::vector<TensorSlice<Index_t, Count_t>> unique_tensor_elements

Subset of all tensor elements. First occurrences of the unique values.

int64_t axis = 0

Axis (optional). Used to gather unique elements over a given dimension.