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.