getOutputShape

abstract fun getOutputShape(inputShape: TensorShape): TensorShape

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

Parameters

inputShape

is a shape of the input.