| 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(@NonNull ArrayType arrayType,
@NonNull INDArray array,
boolean migrateIfInvalid,
boolean exceptionIfDetached) |
LayerWorkspaceMgr.Builder |
LayerWorkspaceMgr.Builder.with(ArrayType type,
String workspaceName,
WorkspaceConfiguration configuration)
Configure the workspace (name, configuration) for the specified array type
|
| Constructor and Description |
|---|
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
Map<ArrayType,WorkspaceConfiguration> configMap,
Map<ArrayType,String> workspaceNames) |
| Modifier and Type | Method and Description |
|---|---|
static INDArray |
ConvolutionUtils.adapt2dMask(INDArray mask,
INDArray output,
@NonNull CNN2DFormat format,
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,
long miniBatchSize,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType) |
static INDArray |
ConvolutionUtils.reshape2dTo4d(INDArray in2d,
long[] toShape,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshape2dTo5d(Convolution3D.DataFormat format,
INDArray in2d,
long n,
long d,
long h,
long w,
long 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,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
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,
CNN2DFormat format,
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.reverseTimeSeries(INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType,
RNNFormat dataFormat) |
static INDArray |
TimeSeriesUtils.reverseTimeSeriesMask(INDArray mask,
LayerWorkspaceMgr workspaceMgr,
ArrayType arrayType)
Reverse a (per time step) time series mask, with shape [minibatch, timeSeriesLength]
|
Copyright © 2021. All rights reserved.