dpuGetOutputTensorInHWCInt8() - 1.2 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2020-07-21
Version
1.2 English

Synopsis

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

Arguments

task
The ctypes pointer to DPU Task.
nodeName
The string DPU Node's name.
size
The size (in Bytes) of output data to be stored.
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 HWC (Height*Width*Channel).

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