singa::LossLayer Class Reference

Base layer for calculating loss and other metrics, e.g., precison. More...

#include <layer.h>

Inheritance diagram for singa::LossLayer:

Public Member Functions
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 void	ComputeFeature (int flag, Metric *perf)=0
	Compute features of this layer based on connected layers. More...
virtual void	ComputeGradient (int flag, Metric *perf)=0
	Compute gradients for parameters and connected layers. 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)

Protected Attributes
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_

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

Detailed Description

Base layer for calculating loss and other metrics, e.g., precison.

Member Function Documentation

Blob<float>* singa::LossLayer::mutable_grad ( const Layer *  from )

inlineoverridevirtual

Returns

a pointer to storing neuron grads of this layer for BP

Reimplemented from 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/layer.h