View source on GitHub |
Computes the shape of a broadcast.
tfp.experimental.distributions.marginal_fns.ps.broadcast_shape(
x_shape, y_shape
)
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.
Returns | |
---|---|
shape
|
A TensorShape or rank-1 integer Tensor representing the
broadcasted shape.
|