Layer

abstract class Layer(name: String)

Base abstract class for all layers.

Parameters

name

Layer name. Would be changed if empty during model compilation.

Constructors

Layer

fun Layer(name: String)

Functions

build

abstract fun build(tf: Ops, kGraph: KGraph, inputShape: Shape)

Extend this function to define variables in layer.

buildFromInboundLayers

fun buildFromInboundLayers(tf: Ops, kGraph: KGraph)

Extend this function to define variables in layer.

computeOutputShape

abstract fun computeOutputShape(inputShape: Shape): Shape

Computes output shape, based on inputShape and Layer type.

computeOutputShapeFromInboundLayers

open fun computeOutputShapeFromInboundLayers(): TensorShape

Computes output shape, based on input shapes of inbound layers.

forward

open fun forward(tf: Ops, input: List<Operand<Float>>, isTraining: Operand<Boolean>, numberOfLosses: Operand<Float>?): Operand<Float>

abstract fun forward(tf: Ops, input: Operand<Float>, isTraining: Operand<Boolean>, numberOfLosses: Operand<Float>?): Operand<Float>

Builds main layer input transformation with tf. Depends on Layer type.

invoke

operator fun invoke(vararg layers: Layer): Layer

Important part of functional API. It takes layers as input and saves them to the inboundLayers of the given layer.

Properties

hasActivation

abstract val hasActivation: Boolean

Returns True, if layer has internal activation function.

inboundLayers

var inboundLayers: MutableList<Layer>

Returns inbound layers.

isTrainable

var isTrainable: Boolean = true

True, if layer's weights could be changed during training. If false, layer's weights are frozen and could be changed during the training.

name

var name: String

outboundLayers

var outboundLayers: MutableList<Layer>

Returns outbound layers.

outputShape

lateinit var outputShape: TensorShape

Output data tensor shape.

paramCount

abstract val paramCount: Int

Returns amount of neurons.

parentModel

var parentModel: TrainableModel? = null

Model where this layer is used.

weights

abstract var weights: Map<String, Array<*>>

Layer's weights.

Inheritors

AbstractActivationLayer

AbstractConv

SeparableConv2D

Dense

Input

AbstractMerge

BatchNorm

AvgPool1D

AvgPool2D

AvgPool3D

GlobalAvgPool1D

GlobalAvgPool2D

GlobalAvgPool3D

GlobalMaxPool1D

GlobalMaxPool2D

GlobalMaxPool3D

MaxPool1D

MaxPool2D

MaxPool3D

Dropout

AbstractCropping

AbstractUpSampling

AbstractZeroPadding

Flatten

Permute

RepeatVector

Reshape