ai.djl.nn.convolutional

Class Conv2dTranspose.Builder

java.lang.Object

ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder<Conv2dTranspose.Builder>

ai.djl.nn.convolutional.Conv2dTranspose.Builder

Enclosing class:

Conv2dTranspose

public static final class Conv2dTranspose.Builder
extends Deconvolution.DeconvolutionBuilder<Conv2dTranspose.Builder>

The Builder to construct a Conv2dTranspose type of Block.

Field Summary

Fields inherited from class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder

dilation, filters, groups, includeBias, kernelShape, outPadding, padding, stride

Method Summary

Modifier and Type	Method and Description
Conv2dTranspose	build() Builds a Conv2dTranspose block.
protected Conv2dTranspose.Builder	self()

Methods inherited from class ai.djl.nn.convolutional.Deconvolution.DeconvolutionBuilder

optBias, optDilation, optGroups, optOutPadding, optPadding, optStride, setFilters, setKernelShape, validate

Methods inherited from class java.lang.Object

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

Method Detail

self

protected Conv2dTranspose.Builder self()

Specified by:

self in class Deconvolution.DeconvolutionBuilder<Conv2dTranspose.Builder>

build

public Conv2dTranspose build()

Builds a Conv2dTranspose block.

Returns:

the Conv2dTranspose block