tfp.experimental.distributions.marginal_fns.ps.rank_from_shape

Computes rank given a Tensor's shape.