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

InitializerOperation

data class InitializerOperation(assign: Operand<Float>, initialValue: Operand<Float>)

A class that keeps information which allows to initialize KVariable.

KVariable

data class KVariable(name: String, shape: Shape, variable: Variable<Float>, initializerOperation: InitializerOperation, regularizer: Regularizer?)

A class that keeps information about a single parameter of the Layer.

Layer

abstract class Layer(name: String)

Base abstract class for all layers.

NoGradients

Marker interface for layers that has no gradients for graph, described in forward method.

ParametrizedLayer

Represents a Layer with parameters.

TrainableLayer

Represents a ParametrizedLayer that can be trained.

freeze

Freezes layer weights, so they won't be changed during training.

initialize

Initializes this layers variables using provided initializer operands.

initializeVariables

Initializes variables for ParametrizedLayer instances using provided initializer operands.

unfreeze

Unfreezes layer weights, allowing to change them during training.

isTrainable

Returns true if this Layer is trainable, false otherwise. Always returns false for layers not implementing TrainableLayer.

paramCount

Returns the number of parameters in this layer. If layer is not a ParametrizedLayer, returns zero.

weights

Weights of this layer. Require parent model to be set on the layer.

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