Method BertModel

example/cpp/bert_model.cpp:238–240

};

BertModel::BertModel(const std::string &filename, DLDeviceType device_type,
                     size_t n_layers, int64_t n_heads)
    : m_(new Impl(filename, device_type, n_layers, n_heads)) {}

std::vector<float> BertModel::operator()(
    const std::vector<std::vector<int64_t>> &inputs,