Trainable Layer
interface TrainableLayer<T : TrainableLayer<T>> : TrainableComponent<T> , Layer<T>
Content copied to clipboard
Functions
extract Tangent
Link copied to clipboard
open override fun extractTangent(output: DTensor, extractor: (DTensor, DTensor) -> DTensor): TrainableComponent.Companion.Tangent
Content copied to clipboard
get Single Input
Link copied to clipboard
Helper to check that the layer was called with a single input. Returns that input if successful, else errors.
load
Link copied to clipboard
store
Link copied to clipboard
training Step
Link copied to clipboard
open override fun trainingStep(optim: Optimizer<*>, tangent: Trainable.Tangent): T
Content copied to clipboard
with Trainables
Link copied to clipboard
Properties
trainables
Link copied to clipboard
Inheritors
Embedding
Link copied to clipboard
EmbeddingBag
Link copied to clipboard
RecurrentBase
Link copied to clipboard
TrainableLayerSingleInput
Link copied to clipboard