Squared Euclidean loss as 0.5 ||predict - ground_truth||^2. More...

#include <loss_layer.h>

Inheritance diagram for singa::EuclideanLossLayer:

Public Member Functions
void	ComputeFeature (int flag, Metric *perf) override
	Compute features of this layer based on connected layers. More...

void	ComputeGradient (int flag, Metric *perf) override
	Compute gradients for parameters and connected layers. More...

Public Member Functions inherited from singa::LossLayer
Blob< float > *	mutable_grad (const Layer *layer) override

const Blob< float > &	grad (const Layer *from) const override

Public Member Functions inherited from singa::Layer
virtual void	Setup (const LayerProto &proto, int npartitions=1)
	Setup layer properties. More...

virtual const std::vector < Param * >	GetParams () const
	Layers that have paramters must override this function. More...

virtual ConnectionType	src_neuron_connection (int k) const
	Return the connection type between one neuron of this layer and its source layer. More...

virtual ConnectionType	dst_layer_connection () const
	Return the connection type of this layer and all dst layers. More...

virtual const std::string	DebugString (int step, int flag)
	For print debug info about each layer, e.g., norm of feature vector, norm of parameters. More...

int	partition_dim () const

int	partition_id () const

int	type () const

const std::string &	name () const
	Return name of this layer.

virtual const Blob< float > &	data (const Layer *from) const

virtual Blob< float > *	mutable_data (const Layer *from)

const std::vector< Layer * >	srclayers () const
	return LayerS that connected to this layer

const std::vector< Layer * >	dstlayers () const
	return LayerS that this layer connected to

int	srclayers_size () const

int	dstlayers_size () const

void	clear_dstlayers ()

void	clear_srclayers ()

void	add_srclayer (Layer *src)

void	add_dstlayer (Layer *dst)

Additional Inherited Members
Static Public Member Functions inherited from singa::Layer
static Layer *	Create (const LayerProto &proto)

Protected Attributes inherited from singa::LossLayer
Blob< float >	metric_

Protected Attributes inherited from singa::Layer
LayerProto	layer_proto_

Blob< float >	data_

Blob< float >	grad_

std::vector< Layer * >	srclayers_

std::vector< Layer * >	dstlayers_

Detailed Description

Squared Euclidean loss as 0.5 ||predict - ground_truth||^2.

Member Function Documentation

void singa::EuclideanLossLayer::ComputeFeature	(	int	flag,
		Metric *	perf
	)

overridevirtual

Compute features of this layer based on connected layers.

Parameters

perf	pointer to Metric obj for collect and aggregate performance

Implements singa::Layer.

void singa::EuclideanLossLayer::ComputeGradient	(	int	flag,
		Metric *	perf
	)

overridevirtual

Compute gradients for parameters and connected layers.

Parameters

flag	used to get the calling phase, e.g., forward of training (kForward \| kTrain)
flag	used to get the calling phase, e.g., forward of training

Implements singa::Layer.

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

/home/wangwei/program/asf/release-0.1/apache-singa-incubating-0.1.0-RC1/include/neuralnet/loss_layer.h

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation