Last updated: 2023-03-16.

tf_quant_finance.utils.get_shape

View source

Returns static shape of x if it is fully defined, or dynamic, otherwise.

tf_quant_finance.utils.get_shape(
    x, name=None
)

####Example

import tensorflow as tf
import tf_quant_finance as tff

x = tf.zeros([5, 2])
prefer_static_shape(x)
# Expected: [5, 2]

#### Args:


* <b>`x`</b>: A tensor of any shape and `dtype`
* <b>`name`</b>: Python string. The name to give to the ops created by this function.
  Default value: `None` which maps to the default name
  `get_shape`.


#### Returns:

A shape of `x` which a list, if the shape is fully defined, or a `Tensor`
for dynamically shaped `x`.