ai.djl.nn.convolutional

Class Conv2D

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.ParameterBlock

ai.djl.nn.convolutional.Convolution

ai.djl.nn.convolutional.Conv2D

All Implemented Interfaces:

Block

public class Conv2D
extends Convolution

Computes 2-D convolution on 4-D input.

The input to a Conv2D is an NDList with a single 4-D NDArray. The layout of the NDArray must be "NCHW". The shapes are

• data: (batch_size, channel, height, width)
• weight: (num_filter, channel, kernel[0], kernel[1])
• bias: (num_filter,)
• out: (batch_size, num_filter, out_height, out_width)
out_height = f(height, kernel[0], pad[0], stride[0], dilate[0])
out_width = f(width, kernel[1], pad[1], stride[1], dilate[1])
where f(x, k, p, s, d) = floor((x + 2 * p - d * (k - 1) - 1)/s) + 1

Both weight and bias are learn-able parameters.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Conv2D.Builder The Builder to construct a Conv2D type of Block.

Nested classes/interfaces inherited from class ai.djl.nn.convolutional.Convolution

Convolution.ConvolutionBuilder<T extends Convolution.ConvolutionBuilder>

Field Summary

Fields inherited from class ai.djl.nn.convolutional.Convolution

bias, dilate, includeBias, kernel, numFilters, numGroups, pad, stride, weight

Fields inherited from class ai.djl.nn.AbstractBlock

inputNames, inputShapes

Method Summary

Modifier and Type	Method and Description
static Conv2D.Builder	builder() Creates a builder to build a Conv2D.
protected LayoutType[]	getExpectedLayout() Returns the expected layout of the input.
protected java.lang.String	getStringLayout() Returns the string representing the layout of the input.
protected int	numDimensions() Returns the number of dimensions of the input.

Methods inherited from class ai.djl.nn.convolutional.Convolution

beforeInitialize, forward, getDirectParameters, getOutputShapes, getParameterShape, loadParameters, saveParameters

Methods inherited from class ai.djl.nn.ParameterBlock

getChildren, initialize, toString

Methods inherited from class ai.djl.nn.AbstractBlock

cast, clear, describeInput, getParameters, isInitialized, readInputShapes, saveInputShapes, setInitializer, setInitializer

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, validateLayout

Method Detail

getExpectedLayout

protected LayoutType[] getExpectedLayout()

Returns the expected layout of the input.

Specified by:

getExpectedLayout in class Convolution

Returns:

the expected layout of the input

getStringLayout

protected java.lang.String getStringLayout()

Returns the string representing the layout of the input.

Specified by:

getStringLayout in class Convolution

Returns:

the string representing the layout of the input

numDimensions

protected int numDimensions()

Returns the number of dimensions of the input.

Specified by:

numDimensions in class Convolution

Returns:

the number of dimensions of the input

builder

public static Conv2D.Builder builder()

Creates a builder to build a Conv2D.

Returns:

a new builder