Attend the Women in ML Symposium on December 7 Register now

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

Stay organized with collections Save and categorize content based on your preferences.

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.