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 java.lang.Object
The Builder to construct aMultiplication
type ofBlock
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Multiplication
build()
Returns the constructedLinear
.Multiplication.Builder
setUnits(long units)
Sets the number of output channels.
-
-
-
Method Detail
-
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 constructedLinear
.- Returns:
- the constructed
Linear
- Throws:
java.lang.IllegalArgumentException
- if all required parameters (outChannels) have not been set
-
-