Method getTensorShape

include/NvInferRuntime.h:1667–1670

\brief Get shape of an input or output tensor. \param tensorName The name of an input or output tensor. \return shape of the tensor, with -1 in place of each dynamic runtime dimension, or Dims{-1, {}} if the provided name does not map to an input or output tensor. \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.

    //! \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.
    //!
    Dims getTensorShape(char const* tensorName) const noexcept
    {
        return mImpl->getTensorShape(tensorName);
    }

    //!
    //! \brief Determine the required data type for a buffer from its binding index.