api/org.diffkt.model/Model

Model

abstract class Model<T : Model<T>> : TrainableComponent<T>

Functions

cpu

open override fun cpu(): T

equals

abstract operator override fun equals(other: Any?): Boolean

extractTangent

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

gpu

open override fun gpu(): T

hashCode

abstract override fun hashCode(): Int

load

open override fun load(from: ByteBuffer): T

predict

open fun predict(data: DTensor): DTensor

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

withLayers

abstract fun withLayers(newLayers: List<Layer<*>>): T

withTrainables

open override 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

layers

abstract val layers: List<Layer<*>>

trainables

open override val trainables: List<Trainable<*>>