Package io.github.kirstenali.deepj.transformer

Record Class TransformerStack

java.lang.Object
java.lang.Record
io.github.kirstenali.deepj.transformer.TransformerStack
All Implemented Interfaces:
Layer, Trainable
public record TransformerStack(List<Layer> blocks) extends Record implements Layer
A sequential stack of transformer blocks.

Each block must implement Layer — forward, backward, and parameters. Supports GPTTransformerBlock, LlamaTransformerBlock, and DeepSeekTransformerBlock blocks, or any custom Layer.

  • Constructor Details

    • TransformerStack

      public TransformerStack(List<Layer> blocks)
      Creates an instance of a TransformerStack record class.
      Parameters:
      blocks - the value for the blocks record component

  • Method Details

    • forward

      public Tensor forward(Tensor x)
      Specified by:
      forward in interface Layer

    • backward

      public Tensor backward(Tensor gradOut)
      Specified by:
      backward in interface Layer

    • parameters

      public List<Parameter> parameters()
      Specified by:
      parameters in interface Trainable

    • toString

      public final String toString()
      Returns a string representation of this record class. The representation contains the name of the class, followed by the name and value of each of the record components.
      Specified by:
      toString in class Record
      Returns:
      a string representation of this object

    • hashCode

      public final int hashCode()
      Returns a hash code value for this object. The value is derived from the hash code of each of the record components.
      Specified by:
      hashCode in class Record
      Returns:
      a hash code value for this object

    • equals

      public final boolean equals(Object o)
      Indicates whether some other object is "equal to" this one. The objects are equal if the other object is of the same class and if all the record components are equal. All components in this record class are compared with Objects::equals(Object,Object).
      Specified by:
      equals in class Record
      Parameters:
      o - the object with which to compare
      Returns:
      true if this object is the same as the o argument; false otherwise.

    • blocks

      public List<Layer> blocks()
      Returns the value of the blocks record component.
      Returns:
      the value of the blocks record component