tfp.experimental.distributions.marginal_fns.ps.tensorshape_util.with_rank

Returns a shape based on x with the given rank.

This method promotes a completely unknown shape to one with a known rank.

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

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

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

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