Method getTensorShape

include/NvInferRuntime.h:2926–2929

\brief Return the shape of the given input or output. \param tensorName The name of an input or output tensor. Return Dims{-1, {}} if the provided name does not map to an input or output tensor. Otherwise return the shape of the input or output tensor. A dimension in an input tensor will have a -1 wildcard value if all the following are true: setInputShape() has not yet been called for this ten

    //! \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 Set values of input tensor required by shape calculations.