ai.djl.nn.convolutional

Class Conv2D.Builder

java.lang.Object

ai.djl.nn.convolutional.Convolution.ConvolutionBuilder<Conv2D.Builder>

ai.djl.nn.convolutional.Conv2D.Builder

Enclosing class:

Conv2D

public static final class Conv2D.Builder
extends Convolution.ConvolutionBuilder<Conv2D.Builder>

The Builder to construct a Conv2D type of Block.

Field Summary

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

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

Constructor Summary

Constructors

Constructor and Description
Builder() Creates a builder that can build a Conv2D block.

Method Summary

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

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

optBias, optDilate, optNumGroups, optPad, optStride, setKernel, setNumFilters, validate

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Creates a builder that can build a Conv2D block.

Method Detail

self

protected Conv2D.Builder self()

Specified by:

self in class Convolution.ConvolutionBuilder<Conv2D.Builder>

build

public Conv2D build()

Builds a Conv2D block.

Returns:

the Conv2D block