api/org.diffkt.model/TrainableComponent

TrainableComponent

interface TrainableComponent<T : TrainableComponent<T>> : Trainable<T>

Types

Companion

object Companion

Functions

cpu

open override fun cpu(): T

extractTangent

open override fun extractTangent(output: DTensor, extractor: (DTensor, DTensor) -> DTensor): TrainableComponent.Companion.Tangent

gpu

open override fun gpu(): T

load

open override fun load(from: ByteBuffer): T

store

open override fun store(into: ByteBuffer): ByteBuffer

to

open fun to(device: Device): OnDevice

trainingStep

open override fun trainingStep(optim: Optimizer<*>, tangent: Trainable.Tangent): T

withTrainables

abstract fun withTrainables(trainables: List<Trainable<*>>): T

wrap

open override fun wrap(wrapper: Wrapper): T

The wrap function should return the same static type it is declared on.

Properties

trainables

abstract val trainables: List<Trainable<*>>

Inheritors

Model

TrainableLayer

TrainableLayerSingleInput