ai.djl.nn.core

Class Linear.Builder

java.lang.Object

ai.djl.nn.core.Linear.Builder

Enclosing class:

Linear

public static final class Linear.Builder
extends java.lang.Object

The Builder to construct a Linear type of Block.

Method Summary

Modifier and Type	Method and 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	optFlatten(boolean flatten) Sets the optional parameter that indicates whether the input tensor should be flattened.
Linear.Builder	setOutChannels(long outChannels) Sets the number of output channels.

Methods inherited from class java.lang.Object

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

Method Detail

setOutChannels

public Linear.Builder setOutChannels(long outChannels)

Sets the number of output channels.

Parameters:

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

optFlatten

public Linear.Builder optFlatten(boolean flatten)

Sets the optional parameter that indicates whether the input tensor should be flattened.

If flatten is set to true, all but the first axis of input data are collapsed together. If false, all but the last axis of input data are kept the same, and the transformation applies on the last axis.

Parameters:

flatten - whether the input tensor should be flattened.

Returns:

this Builder

build

public Linear build()

Returns the constructed Linear.

Returns:

the constructed Linear

Throws:

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