Layer

Functions

cpu

open override fun cpu(): Layer<T>

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(): Layer<T>

invoke

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

open fun to(device: Device): OnDevice

Dropout

LayerSingleInput

TrainableLayer