TrainableLayerSingleInput

Functions

cpu

open override fun cpu(): T

extractTangent

getSingleInput

Helper to check that the layer was called with a single input. Returns that input if successful, else errors.

gpu

open override fun gpu(): T

invoke

abstract operator fun invoke(vararg inputs: DTensor): DTensor

abstract operator fun invoke(input: DTensor): DTensor

load

open override fun load(from: ByteBuffer): T

store

open override fun store(into: ByteBuffer): ByteBuffer

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.

trainables

AffineTransform

BatchNormTrainingBase

Conv2d

Dense

Sequential