tfm.utils.get_shape_list

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

tfm.utils.get_shape_list(
    tensor, expected_rank=None, name=None
)

Args
`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.

Returns
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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-02-02 UTC.

Args