build

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

Extend this function to define variables in layer.

Parameters

tf

TensorFlow graph API for building operations.

kGraph

KGraph to update it.

inputShape

Input shape, result of computeOutputShape call from previous layer.