Package ai.djl.nn.core

Class Linear.Builder

java.lang.Object

ai.djl.nn.core.Linear.Builder

Enclosing class:

Linear

public static class Linear.Builder
extends Object

The Builder to construct a Linear type of Block.

Field Summary

Fields

Modifier and Type	Field	Description
protected long	units

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	Builder()

Method Summary

Modifier and Type	Method	Description
Linear	build()	Returns the constructed Linear.
Linear.Builder	optBias(boolean bias)	Sets the optional parameter that indicates whether to include a bias vector with default value of true.
Linear.Builder	setUnits(long units)	Sets the number of output channels.

Methods inherited from class java.lang.Object

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

Field Details

units

protected long units

Constructor Details

Builder

protected Builder()

Method Details

setUnits

public Linear.Builder setUnits(long units)

Sets the number of output channels.

Parameters:

units - the number of desired output channels

Returns:

this Builder

optBias

public Linear.Builder optBias(boolean bias)

Sets the optional parameter that indicates whether to include a bias vector with default value of true.

Parameters:

bias - whether to use a bias vector parameter

Returns:

this Builder

build

public Linear build()

Returns the constructed Linear.

Returns:

the constructed Linear

Throws:

IllegalArgumentException - if all required parameters (outChannels) have not been set