getOutputShape

open override fun getOutputShape(inputShape: TensorShape): TensorShape
Content copied to clipboard

Returns shape of the output of the operation having input of shape inputShape.

Parameters

inputShape

is a shape of the input.