| Modifier and Type | Method and Description |
|---|---|
protected void |
ComputationGraph.validateArrayWorkspaces(LayerWorkspaceMgr mgr,
INDArray array,
ArrayType arrayType,
String vertexName,
boolean isInputVertex,
String op) |
| Modifier and Type | Method and Description |
|---|---|
protected abstract INDArray |
BaseOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
protected INDArray |
OutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected INDArray |
OCNNOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected INDArray |
RnnOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected INDArray |
CenterLossOutputLayer.getLabels2d(LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
| Modifier and Type | Method and Description |
|---|---|
protected void |
MultiLayerNetwork.validateArrayWorkspaces(LayerWorkspaceMgr mgr,
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 |
|---|---|
INDArray |
LayerWorkspaceMgr.leverageTo(ArrayType arrayType,
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.
|
INDArray |
LayerWorkspaceMgr.validateArrayLocation(ArrayType arrayType,
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 INDArray |
ConvolutionUtils.adapt2dMask(INDArray mask,
INDArray output,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static Pair<INDArray,int[]> |
TimeSeriesUtils.pullLastTimeSteps(INDArray pullFrom,
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 INDArray |
TimeSeriesUtils.reshape2dTo3d(INDArray in,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static INDArray |
ConvolutionUtils.reshape2dTo4d(INDArray in2d,
int[] toShape,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshape2dTo5d(Convolution3D.DataFormat format,
INDArray in2d,
int n,
int d,
int h,
int w,
int ch,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshape3dMask(INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
TimeSeriesUtils.reshape3dTo2d(INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static INDArray |
ConvolutionUtils.reshape4dMask(INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static INDArray |
ConvolutionUtils.reshape4dTo2d(INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshape5dTo2d(Convolution3D.DataFormat format,
INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshapeCnn3dMask(Convolution3D.DataFormat format,
INDArray mask,
INDArray label,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshapeMaskIfRequired(INDArray mask,
INDArray output,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
TimeSeriesUtils.reshapePerOutputTimeSeriesMaskTo2d(INDArray perOutputTimeSeriesMask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static INDArray |
TimeSeriesUtils.reshapeTimeSeriesMaskToCnn4dMask(INDArray timeSeriesMask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reshape time series mask arrays.
|
static INDArray |
TimeSeriesUtils.reshapeTimeSeriesMaskToVector(INDArray timeSeriesMask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reshape time series mask arrays.
|
static INDArray |
TimeSeriesUtils.reverseTimeSeries(INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reverse an input time series along the time dimension
|
static INDArray |
TimeSeriesUtils.reverseTimeSeriesMask(INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reverse a (per time step) time series mask, with shape [minibatch, timeSeriesLength]
|
Copyright © 2019. All rights reserved.