TensorRT
6.0.1.5
|
A Scale layer in a network definition. More...
#include <NvInfer.h>
Public Member Functions | |
virtual void | setMode (ScaleMode mode)=0 |
Set the scale mode. More... | |
virtual ScaleMode | getMode () const =0 |
Set 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... | |
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.
The output size is the same as the input size.
|
pure virtual |
Get the channel axis.
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.
|
pure virtual |
Set the scale mode.
|
pure virtual |
Get the power value.
|
pure virtual |
Get the scale value.
|
pure virtual |
Get the shift value.
|
pure virtual |
Set the scale mode.
|
pure virtual |
Set the power value.
|
pure virtual |
Set the scale value.
|
pure virtual |
Set the shift value.