ct2rs 0.9.18

Rust bindings for OpenNMT/CTranslate2
Documentation 
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
31
32
33
34
35
36
37
38
39
40

#pragma once

#include "ctranslate2/layers/common.h"
#include "ctranslate2/storage_view.h"

namespace ctranslate2 {
  namespace layers {

    // Base class for encoders.
    class Encoder : public Layer {
    public:
      virtual size_t num_input_features() const {
        return 1;
      }

      virtual void operator()(const StorageView& ids,
                              StorageView& output) {
        operator()({ids}, nullptr, output);
      }

      virtual void operator()(const StorageView& ids,
                              const StorageView& lengths,
                              StorageView& output) {
        operator()({ids}, &lengths, output);
      }

      virtual void operator()(const std::vector<StorageView>& ids,
                              const StorageView& lengths,
                              StorageView& output) {
        operator()(ids, &lengths, output);
      }

    protected:
      virtual void operator()(const std::vector<StorageView>& ids,
                              const StorageView* lengths,
                              StorageView& output) = 0;
    };

  }
}