tfp.experimental.auto_batching.TensorType

TensorType(dtype, shape)

dtype A namedtuple alias for field number 0
shape A namedtuple alias for field number 1