Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Macros
Public Member Functions | Protected Attributes | List of all members
singa::LossLayer Class Reference

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

#include <layer.h>

Inheritance diagram for singa::LossLayer:
singa::Layer singa::EuclideanLossLayer singa::SoftmaxLossLayer

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 LayerCreate (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: