std::vector<int64_t> outputShape = outputTensor[0].GetTensorTypeAndShapeInfo().GetShape()
时间: 2023-04-02 18:04:24 浏览: 122
TensorFlow打印输出tensor的值
这个问题是关于 C++ 语言中的 std::vector 类型的问题,我可以回答。这段代码的作用是获取一个张量的形状信息,并将其保存在一个 int64_t 类型的 vector 中。具体来说,outputTensor[] 表示第一个输出张量,GetTensorTypeAndShapeInfo() 函数用于获取张量的类型和形状信息,GetShape() 函数则用于获取张量的形状信息,并将其保存在一个 int64_t 类型的 vector 中。
阅读全文