searchlib/src/vespa/searchlib/tensor/serialized_tensor_ref.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "serialized_tensor_ref.h"

namespace search::tensor {

SerializedTensorRef::SerializedTensorRef()
    : _vectors(),
      _num_mapped_dimensions(0),
      _labels()
{
}

SerializedTensorRef::SerializedTensorRef(VectorBundle vectors, uint32_t num_mapped_dimensions, vespalib::ConstArrayRef<vespalib::string_id> labels)
    : _vectors(vectors),
      _num_mapped_dimensions(num_mapped_dimensions),
      _labels(labels)
{
}

SerializedTensorRef::~SerializedTensorRef() = default;

vespalib::ConstArrayRef<vespalib::string_id>
SerializedTensorRef::get_labels(uint32_t subspace) const
{
    assert(subspace < _vectors.subspaces());
    return {_labels.data() + subspace * _num_mapped_dimensions, _num_mapped_dimensions};
}

}