Is there any common way to check output dimensions of any layer?

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

I am using TensorRT 7.2.2.3 to build a network from scratch.

I found out I can use getNbOutputMaps api to get output dimensions, but i want to check
the output maps for any layer like elementwiselayer, ITensor and sort of.

Is there any way?
Thanks !

#2

Hi,
Please check the below link, as they might answer your concerns
https://docs.nvidia.com/deeplearning/tensorrt/api/index.html
Thanks!