例
dpuGetOutputTensorInHWCFP32
(
task,
nodeName,
int size,
idx = 0
)引数
- task
- DPU タスクへの ctypes 型ポインター。
- nodeName
- 文字列の DPU ノードの名前。
- size
- 格納される出力データのサイズ (バイト単位)。
- idx
- ノードの単一出力テンソルのインデックス。デフォルト値は 0。
説明
DPU タスクの出力テンソル データを INT8 から float32 へ変換し、HWC (Height*Width*Channel) の順序で CPU メモリ バッファーに格納します。
戻り値
出力データを保持する NumPy 配列。エラーの場合、サイズは 0 になります。
インクルード ファイル
n2cube
入力可能なインターフェイス
Vitis AI v1.0