tfp.experimental.distributions.marginal_fns.ps.tensorshape_util.with_rank_at_least

Returns a shape based on x with at least the given rank.

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

x object representing a shape; convertible to tf.TensorShape.
rank An int representing the minimum rank of x or else an assertion is raised.

shape a shape having type(x) but guaranteed to have at least the given rank (or else an assertion was raised).

ValueError If x does not represent a shape with at least the given rank.