Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update SameDiffLambdaLayer.java

Browse Source
This commit is contained in:
agibsonccc 2021-03-09 07:55:15 +09:00
parent 55c52e00b0
commit a3a89fd3ad
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/layers/samediff/SameDiffLambdaLayer.java
View File

@ -30,11 +30,11 @@ import java.util.Map;
public abstract class SameDiffLambdaLayer extends SameDiffLayer { public abstract class SameDiffLambdaLayer extends SameDiffLayer {
/** /**
* The defineLayer method is used to define the foward pass for the layer * The defineLayer method is used to define the forward pass for the layer
* *
* @param sameDiff SameDiff instance to use to define the vertex * @param sameDiff SameDiff instance to use to define the vertex
* @param layerInput Layer input variable * @param layerInput Layer input variable
* @return The output variable (orresponding to the output activations for the layer) * @return The output variable (corresponding to the output activations for the layer)
*/ */
public abstract SDVariable defineLayer(SameDiff sameDiff, SDVariable layerInput); public abstract SDVariable defineLayer(SameDiff sameDiff, SDVariable layerInput);