ActivationLayer

Applies an activation function to an output.

0.2

Constructors

ActivationLayer

fun ActivationLayer(activation: Activations = Activations.Relu, name: String = "")

Creates Dense object.

build

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

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

Extend this function to define variables in the layer and compute layer output.

forward

open override fun forward(tf: Ops, input: Operand<Float>): Operand<Float>

Applies the activation functions to the input to produce the output.

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.

toString

open override fun toString(): String

activation

Activation function.

hasActivation

Returns True, if layer has internal activation function.

inboundLayers

Returns inbound layers.

name

Layer name. A new name is generated during model compilation when provided name is empty.

outboundLayers

Returns outbound layers.

outputShape

Output data tensor shape.

parentModel

Model where this layer is used.