search

NVIDIA / TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
https://developer.nvidia.com/tensorrt
Apache License 2.0
10.61k stars 2.11k forks source link

getOutputDataType understand #3957

Closed lix19937 closed 3 months ago

lix19937 commented 3 months ago

https://docs.nvidia.com/deeplearning/tensorrt/api/c_api/classnvinfer1_1_1_i_plugin_v2_ext.html#af7da001d56d44a42dbba259134dec338

lix19937 commented 3 months ago 

index      | The output tensor index in the valid range between 0 and getNbOutputs()-1.
inputTypes | The data types of the input tensors, stored in an array of length nbInputs.
nbInputs   | The number of input tensors. Will be a non-negative integer.