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.