build

open override fun build(tf: Ops, inputShape: Shape)
Content copied to clipboard

Extend this function to define variables in layer.

Parameters

tf

TensorFlow graph API for building operations.

inputShape

Input shape, result of computeOutputShape call from previous layer.

fun build(tf: Ops)
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.