跳到主要內容
支援知識庫

無法取得有三個輸出的自訂模型的輸出張量

內容類型: 錯誤訊息   |   文章 ID: 000090966   |   最近查看日期: 2026 year 03 month 12 day

說明

  • 自訂模型有三個輸出張量:

    from openvino.runtime import Core
    core = Core()
    model = core.read_model(model="model.xml")
    compiled_model = core.compile_model(model, "CPU")
    infer_request = compiled_model.create_infer_request()
    infer_request.start_async()
    infer_request.wait()
    output = infer_request.get_output_tensor()
    print(output)

  • 收到錯誤:

    RuntimeError: get_output_tensor() must be called on a function with exactly one parameter.

解決方法

無參數的 ov::InferRequest::get_output_tensor 方法可用於只有一個輸出的模型。

  1. 對於有多個輸出的模型,可以用 ov::InferRequest::get_output_tensor方法搭配argument (index: int)

    output1 = infer_request.get_output_tensor(0)
    output2 = infer_request.get_output_tensor(1)
    output3 = infer_request.get_output_tensor(2)

  2. 利用 Tensor 物件的資料屬性來存取推論結果的輸出張量資料。

    output_buffer1 = output2.data
    output_buffer2 = output2.data
    output_buffer3 = output3.data
    print(output_buffer1)
    print(output_buffer2)
    print(output_buffer3)

相關產品

本文章適用 1 個產品。

免責聲明

這個頁面的內容綜合了英文原始內容的人工翻譯譯文與機器翻譯譯文。本內容是基於一般資訊目的,方便您參考而提供,不應視同完整或準確的內容。如果這個頁面的英文版與譯文之間發生任何牴觸,將受英文版規範及管轄。 查看這個頁面的英文版。