tf.broadcast_static_shape

Вычисляет форму расщепления (broadcasting) при известных формах.

tf.broadcast_static_shape(
    shape_x, shape_y
)

Когда shape_x и shape_y полностью известны TensorShape это вычисляет форму, которая представляет собой форму результата операции расщепления (broadcasting) применительно к тензорам с формами shape_x и shape_y.

Например, если shape_x — TensorShape([1, 2, 3]) и shape_y — TensorShape([5, 1, 3]), результат — TensorShape со значением TensorShape([5, 2, 3]).

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

Пример:

shape_x = tf.TensorShape([1, 2, 3])
shape_y = tf.TensorShape([5, 1 ,3])
tf.broadcast_static_shape(shape_x, shape_y)
TensorShape([5, 2, 3])