How can I get output tensor name in python?

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

How can I get output tensor name in python? Thanks

3

And from python sample, it does not need markOutput ?

4

Hi @ding9801,
You can always query the name of a tensor.
You can take reference from the below link for the same.
https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/samplePlugin

Thanks!

5

Thanks Aakanksha.
Is there python sample?

6

Hi @ding9801,
Please find the link here .
You can find all the python sample below.

Thanks!

7

Python sample to get tensor names. How can I get tensor name in python?

8

Hi @ding9801,
Please refer to the below link for python API’s,
https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-721/api/python_api/infer/Graph/LayerBase.html#tensorrt.ITensor

Thanks!