addLayerVariable

fun addLayerVariable(variable: Variable<Float>, isTrainable: Boolean)
Content copied to clipboard

Adds a variable used in layer to the pool of tracked variables.

Parameters

variable

Variable to track in KGraph.

isTrainable

When passed isTrainable = true, the KGraph automatically adds new variables to the tracked collection of graph variables.