Package ai.djl.nn.core

Class Multiplication.Builder

java.lang.Object

ai.djl.nn.core.Multiplication.Builder

Enclosing class:

Multiplication

public static final class Multiplication.Builder
extends Object

The Builder to construct a Multiplication type of Block.

Method Summary

Modifier and Type	Method	Description
Multiplication	build()	Returns the constructed Linear.
Multiplication.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

Method Details

setUnits

public Multiplication.Builder setUnits(long units)

Sets the number of output channels.

Parameters:

units - the number of desired output channels

Returns:

this Builder

build

public Multiplication build()

Returns the constructed Linear.

Returns:

the constructed Linear

Throws:

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