Uses of Class
org.deeplearning4j.nn.workspace.ArrayType
-
-
Uses of ArrayType in org.deeplearning4j.nn.graph
Methods in org.deeplearning4j.nn.graph with parameters of type ArrayType Modifier and Type Method Description protected voidComputationGraph. validateArrayWorkspaces(LayerWorkspaceMgr mgr, INDArray array, ArrayType arrayType, String vertexName, boolean isInputVertex, String op) -
Uses of ArrayType in org.deeplearning4j.nn.layers
Methods in org.deeplearning4j.nn.layers with parameters of type ArrayType Modifier and Type Method Description protected abstract INDArrayBaseOutputLayer. getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)protected INDArrayOutputLayer. getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType) -
Uses of ArrayType in org.deeplearning4j.nn.layers.ocnn
Methods in org.deeplearning4j.nn.layers.ocnn with parameters of type ArrayType Modifier and Type Method Description protected INDArrayOCNNOutputLayer. getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType) -
Uses of ArrayType in org.deeplearning4j.nn.layers.recurrent
Methods in org.deeplearning4j.nn.layers.recurrent with parameters of type ArrayType Modifier and Type Method Description protected INDArrayRnnOutputLayer. getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType) -
Uses of ArrayType in org.deeplearning4j.nn.layers.training
Methods in org.deeplearning4j.nn.layers.training with parameters of type ArrayType Modifier and Type Method Description protected INDArrayCenterLossOutputLayer. getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType) -
Uses of ArrayType in org.deeplearning4j.nn.multilayer
Methods in org.deeplearning4j.nn.multilayer with parameters of type ArrayType Modifier and Type Method Description protected voidMultiLayerNetwork. validateArrayWorkspaces(LayerWorkspaceMgr mgr, INDArray array, ArrayType arrayType, int layerIdx, boolean isPreprocessor, String op) -
Uses of ArrayType in org.deeplearning4j.nn.workspace
Methods in org.deeplearning4j.nn.workspace that return ArrayType Modifier and Type Method Description static ArrayTypeArrayType. 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.Methods in org.deeplearning4j.nn.workspace with parameters of type ArrayType Modifier and Type Method Description INDArrayLayerWorkspaceMgr. leverageTo(ArrayType arrayType, INDArray array)LayerWorkspaceMgr.BuilderLayerWorkspaceMgr.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.INDArrayLayerWorkspaceMgr. validateArrayLocation(@NonNull ArrayType arrayType, @NonNull INDArray array, boolean migrateIfInvalid, boolean exceptionIfDetached)LayerWorkspaceMgr.BuilderLayerWorkspaceMgr.Builder. with(ArrayType type, String workspaceName, WorkspaceConfiguration configuration)Configure the workspace (name, configuration) for the specified array typeConstructor parameters in org.deeplearning4j.nn.workspace with type arguments of type ArrayType Constructor 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) -
Uses of ArrayType in org.deeplearning4j.util
Methods in org.deeplearning4j.util with parameters of type ArrayType Modifier and Type Method Description static INDArrayConvolutionUtils. 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 INDArrayTimeSeriesUtils. reshape2dTo3d(INDArray in, long miniBatchSize, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)static INDArrayConvolutionUtils. reshape2dTo4d(INDArray in2d, long[] toShape, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshape2dTo5d(Convolution3D.DataFormat format, INDArray in2d, long n, long d, long h, long w, long ch, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshape3dMask(INDArray mask, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayTimeSeriesUtils. reshape3dTo2d(INDArray in, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)static INDArrayConvolutionUtils. reshape4dMask(INDArray mask, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)static INDArrayConvolutionUtils. reshape4dTo2d(INDArray in, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshape4dTo2d(INDArray in, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshape5dTo2d(Convolution3D.DataFormat format, INDArray in, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshapeCnn3dMask(Convolution3D.DataFormat format, INDArray mask, INDArray label, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshapeMaskIfRequired(INDArray mask, INDArray output, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayConvolutionUtils. reshapeMaskIfRequired(INDArray mask, INDArray output, LayerWorkspaceMgr workspaceMgr, ArrayType type)static INDArrayTimeSeriesUtils. reshapePerOutputTimeSeriesMaskTo2d(INDArray perOutputTimeSeriesMask, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)static INDArrayTimeSeriesUtils. reshapeTimeSeriesMaskToCnn4dMask(INDArray timeSeriesMask, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)Reshape time series mask arrays.static INDArrayTimeSeriesUtils. reshapeTimeSeriesMaskToVector(INDArray timeSeriesMask, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)Reshape time series mask arrays.static INDArrayTimeSeriesUtils. reverseTimeSeries(INDArray in, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)Reverse an input time series along the time dimensionstatic INDArrayTimeSeriesUtils. reverseTimeSeries(INDArray in, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType, RNNFormat dataFormat)static INDArrayTimeSeriesUtils. reverseTimeSeriesMask(INDArray mask, LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)Reverse a (per time step) time series mask, with shape [minibatch, timeSeriesLength]
-