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

Returns a shape based on x with the given rank.

tfp.experimental.distributions.marginal_fns.ps.tensorshape_util.with_rank(
    x, rank
)

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

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

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

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

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