ai.djl.nn

Class Parameter

java.lang.Object

ai.djl.nn.Parameter

All Implemented Interfaces:

java.lang.AutoCloseable

public class Parameter
extends java.lang.Object
implements java.lang.AutoCloseable

Parameter is a container class that holds a learnable parameter of a model.

Every Parameter is associated with a Block. The output of the block's forward function depends on the values in the Parameter. During training, the values in the Parameter are updated to reflect the training data. This process forms the crux of learning.

Constructor Summary

Constructors

Constructor and Description
Parameter(java.lang.String name, Block block, ParameterType type) Creates a Parameter with the given name, and parameter type, and associated with the given Block.
Parameter(java.lang.String name, Block block, ParameterType type, boolean requiresGrad) Creates a Parameter with the given name, and parameter type, and associated with the given Block.
Parameter(java.lang.String name, Block block, ParameterType type, boolean requireGrad, SparseFormat gradientFormat) Creates a Parameter with the given name, and parameter type, and associated with the given Block.

Method Summary

Modifier and Type	Method and Description
void	close()
NDArray	getArray() Gets the values of this Parameter as an NDArray.
java.lang.String	getId() Gets the ID of this Parameter.
java.lang.String	getName() Gets the name of this Parameter.
ParameterType	getType() Gets the type of this Parameter.
void	initialize(NDManager manager, DataType dataType, Shape[] inputShapes) Initializes the parameter with the given NDManager, with given DataType for the given expected input shapes.
boolean	isInitialized() Checks if this Parameter is initialized.
void	load(NDManager manager, java.io.DataInputStream dis) Loads parameter NDArrays from InputStream.
boolean	requireGradient() Returns whether this parameter needs gradients to be computed.
void	save(java.io.DataOutputStream dos) Writes the parameter NDArrays to the given output stream.
void	setArray(NDArray array) Sets the values of this Parameter.
void	setInitializer(Initializer initializer, boolean overwrite) Sets the Initializer for this Parameter, if not already set.
void	setMandatoryDataType(DataType mandatoryDataType) Sets the mandatory data type for this Parameter.

Methods inherited from class java.lang.Object

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

Constructor Detail

Parameter

public Parameter(java.lang.String name,
                 Block block,
                 ParameterType type)

Creates a Parameter with the given name, and parameter type, and associated with the given Block.

Parameters:

name - the name of the Parameter

block - the block with which this Parameter is associated

type - the type of this Parameter

Parameter

public Parameter(java.lang.String name,
                 Block block,
                 ParameterType type,
                 boolean requiresGrad)

Creates a Parameter with the given name, and parameter type, and associated with the given Block.

Parameters:

name - the name of the Parameter

block - the block with which this Parameter is associated

type - the type of this Parameter

requiresGrad - whether this Parameter needs to compute gradients

Parameter

public Parameter(java.lang.String name,
                 Block block,
                 ParameterType type,
                 boolean requireGrad,
                 SparseFormat gradientFormat)

Creates a Parameter with the given name, and parameter type, and associated with the given Block.

Parameters:

name - the name of the Parameter

block - the block with which this Parameter is associated

type - the type of this Parameter

requireGrad - whether this Parameter needs to compute gradients

gradientFormat - the SparseFormat of the gradient array

Method Detail

getId

public java.lang.String getId()

Gets the ID of this Parameter.

Returns:

the ID of this Parameter

getName

public java.lang.String getName()

Gets the name of this Parameter.

Returns:

the name of this Parameter

getType

public ParameterType getType()

Gets the type of this Parameter.

Returns:

the type of this Parameter

setArray

public void setArray(NDArray array)

Sets the values of this Parameter.

Parameters:

array - the NDArray that contains values of this Parameter

getArray

public NDArray getArray()

Gets the values of this Parameter as an NDArray.

Returns:

an NDArray that contains values of this Parameter

requireGradient

public boolean requireGradient()

Returns whether this parameter needs gradients to be computed.

Returns:

whether this parameter needs gradients to be computed

setMandatoryDataType

public void setMandatoryDataType(DataType mandatoryDataType)

Sets the mandatory data type for this Parameter.

Parameters:

mandatoryDataType - the mandatory data type for this Parameter

isInitialized

public boolean isInitialized()

Checks if this Parameter is initialized.

Returns:

true if this Parameter is initialized

setInitializer

public void setInitializer(Initializer initializer,
                           boolean overwrite)

Sets the Initializer for this Parameter, if not already set. If overwrite flag is true, sets the initializer regardless.

Parameters:

initializer - the initializer to be set

overwrite - if true, set the initializer regardless of whether its already set or not

initialize

public void initialize(NDManager manager,
                       DataType dataType,
                       Shape[] inputShapes)

Initializes the parameter with the given NDManager, with given DataType for the given expected input shapes.

Parameters:

manager - an NDManager to create the arrays

dataType - the datatype of the Parameter

inputShapes - the expected input shapes

save

public void save(java.io.DataOutputStream dos)
          throws java.io.IOException

Writes the parameter NDArrays to the given output stream.

Parameters:

dos - the output stream to write to

Throws:

java.io.IOException - if the write operation fails

load

public void load(NDManager manager,
                 java.io.DataInputStream dis)
          throws java.io.IOException,
                 MalformedModelException

Loads parameter NDArrays from InputStream.

Currently, we cannot deserialize into the exact subclass of NDArray. The SparseNDArray will be loaded as NDArray only.

Parameters:

manager - the NDManager

dis - the InputStream

Throws:

java.io.IOException - if failed to read

MalformedModelException - Exception thrown when model is not in expected format (parameters).

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable