summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorcloudygoose <[email protected]>2015-06-09 21:20:47 +0800
committercloudygoose <[email protected]>2015-06-09 21:20:47 +0800
commitc9326adf0e9f91b87a64f05e9901a38bf64ba4f7 (patch)
tree023289581137f526713837ac82ccd9cc654a6c20
parent3e07a536a89badff65fc81f56bb19cc69291396e (diff)
..
-rw-r--r--doc/nerv_layer.md8
1 files changed, 5 insertions, 3 deletions
diff --git a/doc/nerv_layer.md b/doc/nerv_layer.md
index dd991df..b417729 100644
--- a/doc/nerv_layer.md
+++ b/doc/nerv_layer.md
@@ -15,9 +15,11 @@ __nerv.Layer__ is the base class and most of its methods are abstract.
* __nerv.BiasLayer__ inherits __nerv.Layer__, both `#dim_in` nad `#dim_out` are 1.
* `BiasParam bias` The bias parameter.
* __nerv.SigmoidLayer__ inherits __nerv.Layer__, both `#dim_in` and `#dim_out` are 1.
-* __nerv.SoftmaxCELayer__ inherits __nerv.Layer__, `#dim_in` is 2 and `#dim_out` is 1.
- * `float total_ce`
- * `int total_frams` Records how many frames have passed.
+* __nerv.SoftmaxCELayer__ inherits __nerv.Layer__, `#dim_in` is 2 and `#dim_out` is 0. `input[1]` is the input to the softmax layer, `input[2]` is the reference distribution.
+ * `float total_ce` Records the accumlated cross entropy value.
+ * `int total_frams` Records how many frames have passed.
+ * `bool compressed` The reference distribution can be a one-hot format. This feature is enabled by `layer_conf.compressed`.
+
##Methods##
* __void Layer.\_\_init(Layer self, string id, table global_conf, table layer_conf)__
Abstract method.