ai.djl.nn.convolutional

Class Deconvolution.DeconvolutionBuilder<T extends Deconvolution.DeconvolutionBuilder>

java.lang.Object

ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder<T>

Type Parameters:

T - the type of Deconvolution block to build

Direct Known Subclasses:

Conv1dTranspose.Builder, Conv2dTranspose.Builder

Enclosing class:

Deconvolution

public abstract static class Deconvolution.DeconvolutionBuilder<T extends Deconvolution.DeconvolutionBuilder>
extends java.lang.Object

A builder that can build any Deconvolution block.

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor and Description
DeconvolutionBuilder()

Method Summary

Modifier and Type	Method and Description
T	optBias(boolean includeBias) Sets the optional parameter of whether to include a bias vector.
T	optDilation(Shape dilate) Sets the dilation along each dimension.
T	optGroups(int groups) Sets the number of group partitions.
T	optOutPadding(Shape outPadding) Sets the out_padding along each dimension.
T	optPadding(Shape padding) Sets the padding along each dimension.
T	optStride(Shape stride) Sets the stride of the deconvolution.
protected abstract T	self()
T	setFilters(int filters) Sets the Required number of filters.
T	setKernelShape(Shape kernelShape) Sets the shape of the kernel.
protected void	validate() Validates that the required arguments are set.

Methods inherited from class java.lang.Object

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

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

Constructor Detail

DeconvolutionBuilder

public DeconvolutionBuilder()

Method Detail

setKernelShape

public T setKernelShape(Shape kernelShape)

Sets the shape of the kernel.

Parameters:

kernelShape - the shape of the kernel

Returns:

this Builder

optStride

public T optStride(Shape stride)

Sets the stride of the deconvolution. Defaults to 1 in each dimension.

Parameters:

stride - the shape of the stride

Returns:

this Builder

optPadding

public T optPadding(Shape padding)

Sets the padding along each dimension. Defaults to 0 along each dimension.

Parameters:

padding - the shape of padding along each dimension

Returns:

this Builder

optOutPadding

public T optOutPadding(Shape outPadding)

Sets the out_padding along each dimension. Defaults to 0 along each dimension.

Parameters:

outPadding - the shape of out_padding along each dimension

Returns:

this Builder

optDilation

public T optDilation(Shape dilate)

Sets the dilation along each dimension. Defaults to 1 along each dimension.

Parameters:

dilate - the shape of dilation along each dimension

Returns:

this Builder

setFilters

public T setFilters(int filters)

Sets the Required number of filters.

Parameters:

filters - the number of deconvolution filters(channels)

Returns:

this Builder

optGroups

public T optGroups(int groups)

Sets the number of group partitions.

Parameters:

groups - the number of group partitions

Returns:

this Builder

optBias

public T optBias(boolean includeBias)

Sets the optional parameter of whether to include a bias vector. Includes bias by default.

Parameters:

includeBias - whether to use a bias vector parameter

Returns:

this Builder

validate

protected void validate()

Validates that the required arguments are set.

Throws:

java.lang.IllegalArgumentException - if the required arguments are not set

self

protected abstract T self()