ai.djl.nn

Class Parameter.Builder

java.lang.Object

ai.djl.nn.Parameter.Builder

Enclosing class:

Parameter

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

A Builder to construct a Parameter.

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
Parameter	build() Builds a Parameter instance.
Parameter.Builder	optArray(NDArray array) Sets the array of the Parameter.
Parameter.Builder	optInitializer(Initializer initializer) Sets the Initializer of the Parameter.
Parameter.Builder	optRequiresGrad(boolean requiresGrad) Sets if the Parameter requires gradient.
Parameter.Builder	optShape(Shape shape) Sets the shape of the Parameter.
Parameter.Builder	setName(java.lang.String name) Sets the name of the Parameter.
Parameter.Builder	setType(Parameter.Type type) Sets the Type of the Parameter.

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

setName

public Parameter.Builder setName(java.lang.String name)

Sets the name of the Parameter.

Parameters:

name - the name of the Parameter

Returns:

this Parameter

setType

public Parameter.Builder setType(Parameter.Type type)

Sets the Type of the Parameter.

Parameters:

type - the Type of the Parameter

Returns:

this Parameter

optShape

public Parameter.Builder optShape(Shape shape)

Sets the shape of the Parameter.

Parameters:

shape - the shape of the Parameter

Returns:

this Parameter

optInitializer

public Parameter.Builder optInitializer(Initializer initializer)

Sets the Initializer of the Parameter.

Parameters:

initializer - the Initializer of the Parameter

Returns:

this Parameter

optArray

public Parameter.Builder optArray(NDArray array)

Sets the array of the Parameter.

Parameters:

array - the array of the Parameter

Returns:

this Parameter

optRequiresGrad

public Parameter.Builder optRequiresGrad(boolean requiresGrad)

Sets if the Parameter requires gradient.

Parameters:

requiresGrad - if the Parameter requires gradient

Returns:

this Parameter

build

public Parameter build()

Builds a Parameter instance.

Returns:

the Parameter instance