Trainable

Types

Tangent

cpu

abstract override fun cpu(): T

extractTangent

abstract fun extractTangent(output: DTensor, extractor: (input: DTensor, output: DTensor) -> DTensor): Trainable.Tangent

gpu

abstract override fun gpu(): T

load

abstract fun load(from: ByteBuffer): T

store

open fun to(device: Device): OnDevice

trainingStep

wrap

abstract fun wrap(wrapper: Wrapper): T

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

TrainableComponent

TrainableTensor