Package ai.djl.nn.norm

Class BatchNorm.Builder

java.lang.Object

ai.djl.nn.norm.BatchNorm.BaseBuilder<BatchNorm.Builder>

ai.djl.nn.norm.BatchNorm.Builder

Enclosing class:

BatchNorm

public static class BatchNorm.Builder
extends BatchNorm.BaseBuilder<BatchNorm.Builder>

The Builder to construct a BatchNorm.

Field Summary

Fields inherited from class ai.djl.nn.norm.BatchNorm.BaseBuilder

axis, center, epsilon, momentum, scale

Method Summary

Modifier and Type	Method	Description
BatchNorm	build()	Builds the new BatchNorm.
BatchNorm.Builder	self()	Returns this {code Builder} object.

Methods inherited from class ai.djl.nn.norm.BatchNorm.BaseBuilder

optAxis, optCenter, optEpsilon, optMomentum, optScale

Methods inherited from class java.lang.Object

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

Method Details

build

public BatchNorm build()

Builds the new BatchNorm.

Specified by:

build in class BatchNorm.BaseBuilder<BatchNorm.Builder>

Returns:

the new BatchNorm

self

public BatchNorm.Builder self()

Returns this {code Builder} object.

Specified by:

self in class BatchNorm.BaseBuilder<BatchNorm.Builder>

Returns:

this BaseBuilder