tfp.experimental.distributions.marginal_fns.ps.broadcast_shape

Computes the shape of a broadcast.

When both arguments are statically-known, the broadcasted shape will be computed statically and returned as a TensorShape. Otherwise, a rank-1 Tensor will be returned.

x_shape A TensorShape or rank-1 integer Tensor. The input Tensor is broadcast against this shape.
y_shape A TensorShape or rank-1 integer Tensor. The input Tensor is broadcast against this shape.

shape A TensorShape or rank-1 integer Tensor representing the broadcasted shape.