我想知道如何使用tensorflow C ++ API:tensorflow::ops::SaveV2
进行检查点。它的签名是:
SaveV2(
const ::tensorflow::Scope & scope,::tensorflow::Input prefix,::tensorflow::Input tensor_names,::tensorflow::Input shape_and_slices,::tensorflow::InputList tensors
)
我的前缀是c ++字符串,我想保存的不同张量(Tensor
对象)包括其名称(std::vector<string> names
)和shapes(TensorShape
对象) 。我的问题是如何将它们转换为签名中的那些不同类型。作用域参数没有问题。我的问题在于前缀,张量名称,shapes_and_slices和张量。如果有人可以编写示例代码块以列出的类型实现上述例程,我会很高兴。