LayerSingleInput

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

open operator override fun invoke(vararg inputs: DTensor): DTensor

abstract operator fun invoke(input: DTensor): DTensor

open fun to(device: Device): OnDevice

Activation

AvgPool2d

Flatten

MaxPool2d

ReluLayer

TrainableLayerSingleInput