from tensorflow.python.framework import tensor_shape as _tensor_shape

Input:

Output:

Prompt: