getOutputShape

open override fun getOutputShape(inputShape: TensorShape): TensorShape