c++ - 如何使用protobuf序列化元组（std :: string，torch :: Tensor）的libtorch向量？

Question

我应该如何定义 ProtoBuf 模式以便我可以序列化这个形状向量？

std::vector<std::tuple(std::string, torch::Tensor)>

我能找到的只是简单的琐碎示例。我不知道应该如何在 ProtoBuf 模式中定义torch::Tensoror ！tuple

Answer 1

您可以在 ProtoBuf 模式中使用string来表示。torch::Tensor

例子：

syntax = "proto3";

package tensor;

message Lookup {
    message Tuple {
        string key = 1;
        string tensor = 2;
    }

    repeated Tuple tuples = 1;
}

使用torch::save()和torch::load()API 转换为/从这里提到的std::stringvia 。std::stringstream