Method setAxis

include/NvInfer.h:6358–6361

\brief Set the axis used by ScatterMode::kELEMENTS. The axis defaults to 0.

    //! The axis defaults to 0.
    //!
    void setAxis(int32_t axis) noexcept
    {
        mImpl->setAxis(axis);
    }

    //!
    //! \brief Get the axis.