c - 如何使用Tensorflow的C api从TF_Operation获取TF_Input？

Question

我正在使用 Tensorflow 1.13.1 的 C-API。

通过 TF_GraphOperationByName 获取 TF_Operation 后，我找不到获取 TF_Input 或 TF_Output 的方法，而我想获取其中的数据类型。

以下是相关方法 ( https://github.com/tensorflow/tensorflow/blob/r1.13/tensorflow/c/c_api.h )

TF_CAPI_EXPORT extern int TF_OperationNumInputs(TF_Operation* oper);

TF_CAPI_EXPORT extern TF_DataType TF_OperationInputType(TF_Input oper_in);

在我看来，它缺少像“TF_Input TF_OperationInput(int32_t index);”这样的方法

TF_Operation *input_op = TF_GraphOperationByName(graph, "Placeholder");
int input_num = TF_OperationNumInputs(input_op);
for (int i = 0; i < input_num; ++i) {
  // Expectation: auto oper_in = TF_OperationInput(i);
  auto data_type = TF_OperationInputType(oper_in);
}

请帮助我找到一种从每个 TF_Input 和 TF_Output 获取 TF_Datatype 的方法。

非常感谢。

Answer 1

我认为您正在寻找TF_OperationInput() ( link )。

此方法接收 TF_Input ( link ) 作为参数，您可以使用 { input_op, i } 构建该参数。