build

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

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

Parameters

tf

TensorFlow graph API for building operations.

input

Layer input.

isTraining

TensorFlow operand for switching between training and inference modes.

numberOfLosses

TensorFlow operand for batch size data.

fun build(tf: Ops): Placeholder<Float>
Content copied to clipboard

Extend this function to define placeholder in layer.

NOTE: Called instead of Layer.build.

Parameters

tf

TensorFlow graph API for building operations.