tf.broadcast_dynamic_shape

Вычисляет форму результата широковещательного распространения по заданным символическим формам.

tf.broadcast_dynamic_shape(
    shape_x, shape_y
)

Если shape_x и shape_y — тензоры, представляющие формы (то есть результат вызова tf.shape для другого тензора), эта функция вычисляет тензор, содержащий форму результата операции широковещательного распространения, применённой к тензорам с формами shape_x и shape_y.

Это полезно для проверки результата операции широковещательного распространения, если статические формы тензоров неизвестны.

Пример:

shape_x = (1, 2, 3)
shape_y = (5, 1, 3)
tf.broadcast_dynamic_shape(shape_x, shape_y)
<tf.Tensor: shape=(3,), dtype=int32, numpy=array([5, 2, 3], ...>