An Activation layer in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IActivationLayer:

Public Member Functions
virtual void	setActivationType (ActivationType type)=0
	Set the type of activation to be performed. More...

virtual ActivationType	getActivationType () const =0
	Get the type of activation to be performed. 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...

Detailed Description

An Activation layer in a network definition.

This layer applies a per-element activation function to its input.

The output has the same shape as the input.

Member Function Documentation

virtual ActivationType nvinfer1::IActivationLayer::getActivationType ( ) const

pure virtual

Get the type of activation to be performed.

virtual void nvinfer1::IActivationLayer::setActivationType ( ActivationType type )

pure virtual

Set the type of activation to be performed.

The documentation for this class was generated from the following file: