getOutputShape

open override fun getOutputShape(inputShape: TensorShape): TensorShape

Takes result color mode into account when computing output shape.