tfp.experimental.distributions.marginal_fns.ps.tensorshape_util.dims

Returns a list of dimension sizes, or None if rank is unknown.

For more details, see help(tf.TensorShape.dims).

x object representing a shape; convertible to tf.TensorShape.

shape_as_list list of sizes or None values representing each dimensions size if known. A size is tf.Dimension if input is a tf.TensorShape and an int otherwise.