Package 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 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

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

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	Builder()	Creates a builder that can build a Conv2d block.

Method Summary

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

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

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

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

protected Builder()

Creates a builder that can build a Conv2d block.

Method Details

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