Trainable

All Known Subinterfaces:: Layer

All Known Implementing Classes:: Embedding, FNN, GPTModel, LayerNorm1D, Linear, MultiHeadSelfAttention, PositionalEmbedding, TransformerBlock, TransformerStack

public interface Trainable

Method Summary

Modifier and Type

Method

Description

List<Parameter>

parameters()

default void

zeroGrad()

Method Details
- parameters
  
  List<Parameter> parameters()
- zeroGrad
  
  default void zeroGrad()