Package ai.djl.nn

Class Parameter.Builder

java.lang.Object
ai.djl.nn.Parameter.Builder
Enclosing class:
Parameter
public static final class Parameter.Builder extends Object
A Builder to construct a Parameter.

  • Constructor Details

    • Builder

      public Builder()

  • Method Details

    • setName

      public Parameter.Builder setName(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