Package | Description |
---|---|
org.deeplearning4j.nn.api | |
org.deeplearning4j.nn.graph | |
org.deeplearning4j.nn.multilayer |
Modifier and Type | Method and Description |
---|---|
static FwdPassType |
FwdPassType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static FwdPassType[] |
FwdPassType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
protected Map<String,INDArray> |
ComputationGraph.ffToLayerActivationsDetached(boolean train,
@NonNull FwdPassType fwdPassType,
boolean storeLastForTBPTT,
int layerIndex,
int[] excludeIdxs,
@NonNull INDArray[] features,
INDArray[] fMask,
INDArray[] lMask,
boolean clearLayers)
Feed-forward through the network - returning all array activations detached from any workspace.
|
protected Map<String,INDArray> |
ComputationGraph.ffToLayerActivationsInWS(boolean train,
int layerIndex,
int[] excludeIdxs,
FwdPassType fwdPassType,
boolean storeLastForTBPTT,
INDArray[] input,
INDArray[] fMask,
INDArray[] lMask,
boolean clearInputs)
Feed-forward through the network - if workspaces are used, all returned activations will be present in workspace
WS_ALL_LAYERS_ACT.
|
protected INDArray[] |
ComputationGraph.outputOfLayersDetached(boolean train,
@NonNull FwdPassType fwdPassType,
@NonNull int[] layerIndexes,
@NonNull INDArray[] features,
INDArray[] fMask,
INDArray[] lMasks,
boolean clearLayerInputs,
boolean detachedInputs,
MemoryWorkspace outputWorkspace)
Provide the output of the specified layers, detached from any workspace.
|
Modifier and Type | Method and Description |
---|---|
protected List<INDArray> |
MultiLayerNetwork.ffToLayerActivationsDetached(boolean train,
@NonNull FwdPassType fwdPassType,
boolean storeLastForTBPTT,
int layerIndex,
@NonNull INDArray input,
INDArray fMask,
INDArray lMask,
boolean clearInputs)
Feed-forward through the network - returning all array activations in a list, detached from any workspace.
|
protected List<INDArray> |
MultiLayerNetwork.ffToLayerActivationsInWs(int layerIndex,
@NonNull FwdPassType fwdPassType,
boolean storeLastForTBPTT,
@NonNull INDArray input,
INDArray fMask,
INDArray lMask)
Feed-forward through the network at training time - returning a list of all activations in a workspace (WS_ALL_LAYERS_ACT)
if workspaces are enabled for training; or detached if no workspaces are used.
|
protected INDArray |
MultiLayerNetwork.outputOfLayerDetached(boolean train,
@NonNull FwdPassType fwdPassType,
int layerIndex,
@NonNull INDArray input,
INDArray featureMask,
INDArray labelsMask,
MemoryWorkspace outputWorkspace)
Provide the output of the specified layer, detached from any workspace.
|
Copyright © 2020. All rights reserved.