Method addScaleNd

include/NvInfer.h:7857–7861

\brief Add a multi-dimension scale layer to the network. \param input The input tensor to the layer. \param mode The scaling mode. \param shift The shift value. \param scale The scale value. \param power The power value. \param channelAxis The channel axis. If the weights are available, then the size of weights are dependent on the ScaleMode. For ScaleMode::kUNIFORM, the number of weights equals

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

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