tfm.utils.get_shape_list

Returns a list of the shape of tensor, preferring static dimensions.

tensor A tf.Tensor object to find the shape of.
expected_rank (optional) int. The expected rank of tensor. If this is specified and the tensor has a different rank, and exception will be thrown.
name Optional name of the tensor for the error message.

A list of dimensions of the shape of tensor. All static dimensions will be returned as python integers, and dynamic dimensions will be returned as tf.Tensor scalars.