Interface | Description |
---|---|
LayerHelper |
Class | Description |
---|---|
AbstractLayer<LayerConfT extends Layer> |
A layer with input and output, no parameters or gradients
|
ActivationLayer | |
BaseLayer<LayerConfT extends BaseLayer> |
A layer with parameters
|
BaseOutputLayer<LayerConfT extends BaseOutputLayer> | |
BasePretrainNetwork<LayerConfT extends BasePretrainNetwork> | |
DropoutLayer | |
FrozenLayer | |
FrozenLayerWithBackprop |
Frozen layer freezes parameters of the layer it wraps, but allows the backpropagation to continue.
|
HelperUtils |
Simple meta helper util class for instantiating
platform specific layer helpers that handle interaction with
lower level libraries like cudnn and onednn.
|
LossLayer | |
OutputLayer | |
RepeatVector |
Copyright © 2022. All rights reserved.