A Scale layer in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IScaleLayer:

Public Member Functions
virtual void	setMode (ScaleMode mode)=0
	Set the scale mode. More...

virtual ScaleMode	getMode () const =0
	Get the scale mode. More...

virtual void	setShift (Weights shift)=0
	Set the shift value. More...

virtual Weights	getShift () const =0
	Get the shift value. More...

virtual void	setScale (Weights scale)=0
	Set the scale value. More...

virtual Weights	getScale () const =0
	Get the scale value. More...

virtual void	setPower (Weights power)=0
	Set the power value. More...

virtual Weights	getPower () const =0
	Get the power value. More...

virtual int	getChannelAxis () const =0
	Get the channel axis. More...

Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...

virtual void	setName (const char *name)=0
	Set the name of a layer. More...

virtual const char *	getName () const =0
	Return the name of a layer. More...

virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.

virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...

virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.

virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...

virtual void	setInput (int index, ITensor &tensor)=0
	Replace an input of this layer with a specific tensor. More...

virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...

virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...

virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...

virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...

virtual void	setOutputType (int index, DataType dataType)=0
	Set the output type of this layer. More...

virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...

virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...

virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

Detailed Description

A Scale layer in a network definition.

This layer applies a per-element computation to its input:

output = (input* scale + shift)^ power

The coefficients can be applied on a per-tensor, per-channel, or per-element basis.

Note: If the number of weights is 0, then a default value is used for shift, power, and scale. The default shift is 0, the default power is 1, and the default scale is 1.

The output size is the same as the input size.

Note: The input tensor for this layer is required to have a minimum of 3 dimensions.

See also: ScaleMode

Warning: Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

virtual int nvinfer1::IScaleLayer::getChannelAxis ( ) const

pure virtual

Get the channel axis.

Returns: channelAxis parameter passed to addScaleNd()

The value is the index of the channel axis in the input tensor's dimensions. All dimensions after the channel axis are assumed to be spatial dimensions, and the only spatial dimensions in the tensor. The number of spatial dimensions is thus getDimensions().nbDims - channelAxis - 1. Supported numbers of spatial dimensions are 2 and 3 for 2d and 3d scale layers respectively.