| Modifier and Type | Method and Description |
|---|---|
protected void |
ComputationGraph.validateArrayWorkspaces(LayerWorkspaceMgr mgr,
org.nd4j.linalg.api.ndarray.INDArray array,
ArrayType arrayType,
String vertexName,
boolean isInputVertex,
String op) |
| Modifier and Type | Method and Description |
|---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
OutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
protected abstract org.nd4j.linalg.api.ndarray.INDArray |
BaseOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
OCNNOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
RnnOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
CenterLossOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected void |
MultiLayerNetwork.validateArrayWorkspaces(LayerWorkspaceMgr mgr,
org.nd4j.linalg.api.ndarray.INDArray array,
ArrayType arrayType,
int layerIdx,
boolean isPreprocessor,
String op) |
| Modifier and Type | Method and Description |
|---|---|
static ArrayType |
ArrayType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static ArrayType[] |
ArrayType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
org.nd4j.linalg.api.ndarray.INDArray |
LayerWorkspaceMgr.leverageTo(ArrayType arrayType,
org.nd4j.linalg.api.ndarray.INDArray array) |
LayerWorkspaceMgr.Builder |
LayerWorkspaceMgr.Builder.noWorkspaceFor(ArrayType type)
Specify that no workspace should be used for array of the specified type - i.e., these arrays should all
be scoped out.
|
org.nd4j.linalg.api.ndarray.INDArray |
LayerWorkspaceMgr.validateArrayLocation(ArrayType arrayType,
org.nd4j.linalg.api.ndarray.INDArray array,
boolean migrateIfInvalid,
boolean exceptionIfDetached) |
LayerWorkspaceMgr.Builder |
LayerWorkspaceMgr.Builder.with(ArrayType type,
String workspaceName,
org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration)
Configure the workspace (name, configuration) for the specified array type
|
| Constructor and Description |
|---|
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
| Modifier and Type | Method and Description |
|---|---|
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.adapt2dMask(org.nd4j.linalg.api.ndarray.INDArray mask,
org.nd4j.linalg.api.ndarray.INDArray output,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.ndarray.INDArray,int[]> |
TimeSeriesUtils.pullLastTimeSteps(org.nd4j.linalg.api.ndarray.INDArray pullFrom,
org.nd4j.linalg.api.ndarray.INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Extract out the last time steps (2d array from 3d array input) accounting for the mask layer, if present.
|
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reshape2dTo3d(org.nd4j.linalg.api.ndarray.INDArray in,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.reshape2dTo4d(org.nd4j.linalg.api.ndarray.INDArray in2d,
int[] toShape,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.reshape3dMask(org.nd4j.linalg.api.ndarray.INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reshape3dTo2d(org.nd4j.linalg.api.ndarray.INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.reshape4dMask(org.nd4j.linalg.api.ndarray.INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.reshape4dTo2d(org.nd4j.linalg.api.ndarray.INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionUtils.reshapeMaskIfRequired(org.nd4j.linalg.api.ndarray.INDArray mask,
org.nd4j.linalg.api.ndarray.INDArray output,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reshapePerOutputTimeSeriesMaskTo2d(org.nd4j.linalg.api.ndarray.INDArray perOutputTimeSeriesMask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reshapeTimeSeriesMaskToVector(org.nd4j.linalg.api.ndarray.INDArray timeSeriesMask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reshape time series mask arrays.
|
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reverseTimeSeries(org.nd4j.linalg.api.ndarray.INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reverse an input time series along the time dimension
|
static org.nd4j.linalg.api.ndarray.INDArray |
TimeSeriesUtils.reverseTimeSeriesMask(org.nd4j.linalg.api.ndarray.INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reverse a (per time step) time series mask, with shape [minibatch, timeSeriesLength]
|
Copyright © 2018. All rights reserved.