ai.djl.nn.convolutional

Class Deconvolution

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.convolutional.Deconvolution

All Implemented Interfaces:

Block

Direct Known Subclasses:

Conv1dTranspose, Conv2dTranspose

public abstract class Deconvolution
extends AbstractBlock

Transposed convolution, also named fractionally-strided convolution Dumoulin & Visin or deconvolution Long et al., 2015, serves this purpose.

The need for transposed convolutions generally arises from the desire to use a transformation going in the opposite direction of a normal convolution, i.e., from something that has the shape of the output of some convolution to something that has the shape of its input while maintaining a connectivity pattern that is compatible with said convolution.

Current implementations of Deconvolution are Conv1dTranspose with input dimension of LayoutType.WIDTH and Conv2dTranspose with input dimension of LayoutType.WIDTH and LayoutType.HEIGHT. These implementations share the same core principal as a Deconvolution layer does, with the difference being the number of input dimension each operates on as denoted by ConvXdTranspose for X dimension(s).

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Deconvolution.DeconvolutionBuilder<T extends Deconvolution.DeconvolutionBuilder> A builder that can build any Deconvolution block.

Field Summary

Fields

Modifier and Type	Field and Description
protected Parameter	bias
protected Shape	dilation
protected int	filters
protected int	groups
protected boolean	includeBias
protected Shape	kernelShape
protected Shape	outPadding
protected Shape	padding
protected Shape	stride
protected Parameter	weight

Fields inherited from class ai.djl.nn.AbstractBlock

children, inputNames, inputShapes, parameters, parameterShapeCallbacks, version

Constructor Summary

Constructors

Constructor and Description
Deconvolution(Deconvolution.DeconvolutionBuilder<?> builder) Creates a Deconvolution object.

Method Summary

Modifier and Type	Method and Description
protected void	beforeInitialize(Shape[] inputs) Performs any action necessary before initialization.
protected NDList	forwardInternal(ParameterStore parameterStore, NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)
protected abstract LayoutType[]	getExpectedLayout() Returns the expected layout of the input.
Shape[]	getOutputShapes(NDManager manager, Shape[] inputs) Returns the expected output shapes of the block for the specified input shapes.
protected abstract java.lang.String	getStringLayout() Returns the string representing the layout of the input.
void	loadMetadata(byte version, java.io.DataInputStream is) Overwrite this to load additional metadata with the parameter values.
protected abstract int	numDimensions() Returns the number of dimensions of the input.

Methods inherited from class ai.djl.nn.AbstractBlock

addChildBlock, addParameter, addParameter, addParameter, cast, clear, describeInput, forward, getChildren, getDirectParameters, getParameters, getParameterShape, initialize, initializeChildBlocks, isInitialized, loadParameters, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.nn.Block

forward, forward, validateLayout

Field Detail

kernelShape

protected Shape kernelShape

stride

protected Shape stride

padding

protected Shape padding

outPadding

protected Shape outPadding

dilation

protected Shape dilation

filters

protected int filters

groups

protected int groups

includeBias

protected boolean includeBias

weight

protected Parameter weight

bias

protected Parameter bias

Constructor Detail

Deconvolution

public Deconvolution(Deconvolution.DeconvolutionBuilder<?> builder)

Creates a Deconvolution object.

Parameters:

builder - the Builder that has the necessary configurations

Method Detail

getExpectedLayout

protected abstract LayoutType[] getExpectedLayout()

Returns the expected layout of the input.

Returns:

the expected layout of the input

getStringLayout

protected abstract java.lang.String getStringLayout()

Returns the string representing the layout of the input.

Returns:

the string representing the layout of the input

numDimensions

protected abstract int numDimensions()

Returns the number of dimensions of the input.

Returns:

the number of dimensions of the input

forwardInternal

protected NDList forwardInternal(ParameterStore parameterStore,
                                 NDList inputs,
                                 boolean training,
                                 ai.djl.util.PairList<java.lang.String,java.lang.Object> params)

Specified by:

forwardInternal in class AbstractBlock

beforeInitialize

protected void beforeInitialize(Shape[] inputs)

Performs any action necessary before initialization.

Overrides:

beforeInitialize in class AbstractBlock

Parameters:

inputs - the expected shapes of the input

getOutputShapes

public Shape[] getOutputShapes(NDManager manager,
                               Shape[] inputs)

Returns the expected output shapes of the block for the specified input shapes.

Parameters:

manager - an NDManager

inputs - the shapes of the inputs

Returns:

the expected output shapes of the block

loadMetadata

public void loadMetadata(byte version,
                         java.io.DataInputStream is)
                  throws java.io.IOException,
                         MalformedModelException

Overwrite this to load additional metadata with the parameter values.

If you overwrite AbstractBlock.saveMetadata(DataOutputStream) or need to provide backward compatibility to older binary formats, you prabably need to overwrite this. This default implementation checks if the version number fits, if not it throws an MalformedModelException. After that it restores the input shapes.

Overrides:

loadMetadata in class AbstractBlock

Parameters:

version - the version used for loading this metadata.

is - the input stream we are loading from

Throws:

java.io.IOException - loading failed

MalformedModelException - data can be loaded but has wrong format