dpuGetOutputTensorInCHWInt8() - 1.3 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-02-03
Version
1.3 English

Synopsis

dpuGetOutputTensorInCHWInt8
(
task,
nodeName,
int size,
idx = 0
    )

Arguments

task
The ctypes pointer to DPU Task.
size
The string DPU Node's name.
idx
The index of a single output tensor for the Node, with default value of 0.

Description

Get DPU Task’s output Tensor and store its INT8 type data into CPU memory buffer under the layout of CHW (Channel*Height*Width).

Returns

NumPy array to hold the output data. Its size is zero in case of any error.

Include File

n2cube

Availability

Vitis AI v1.0