Package org.jetbrains.kotlinx.dl.api.core.layer.core

Types

ActivationLayer
Link copied to clipboard
class ActivationLayer(activation: Activations, name: String) : AbstractActivationLayer

Applies an activation function to an output.

Dense
Link copied to clipboard
class Dense(outputSize: Int, activation: Activations, kernelInitializer: Initializer, biasInitializer: Initializer, kernelRegularizer: Regularizer?, biasRegularizer: Regularizer?, activityRegularizer: Regularizer?, useBias: Boolean, name: String) : Layer, TrainableLayer

Densely-connected (fully-connected) layer class.

Input
Link copied to clipboard
class Input(dims: Long, name: String) : Layer

This layer is responsible for the input shape of the built model.