Method addScale

include/NvInfer.h:7049–7052

\brief Add a Scale layer to the network. \param input The input tensor to the layer. This tensor is required to have a minimum of 3 dimensions in implicit batch mode and a minimum of 4 dimensions in explicit batch mode. \param mode The scaling mode. \param shift The shift value. \param scale The scale value. \param power The power value. If the weights are available, then the size of weights are

    //! \return The new Scale layer, or nullptr if it could not be created.
    //!
    IScaleLayer* addScale(ITensor& input, ScaleMode mode, Weights shift, Weights scale, Weights power) noexcept
    {
        return mImpl->addScale(input, mode, shift, scale, power);
    }

    //!
    //! \brief Add a SoftMax layer to the network.