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

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

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